码迷,mamicode.com
首页 > 其他好文 > 详细

矩阵快速幂模板

时间:2020-05-24 00:46:48      阅读:56      评论:0      收藏:0      [点我收藏+]

标签:整数   end   代码   html   lse   col   com   i++   space   

先贴个整数快速幂

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll b,p,k,ans=1,res;

int main(){
    scanf("%lld%lld%lld",&b,&p,&k);
    cout<<b<<"^"<<p<<" mod "<<k<<"=";
    while(p){
        if(p&1)ans=ans*b%k;
        b=b*b%k;
        p=p>>1;
    }
    cout<<ans%k<<endl;
}

矩阵快速幂https://www.cnblogs.com/cmmdc/p/6936196.html

代码如下

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const int maxn=1e2+5;
const ll mod= 1e9+7;
struct matrix{
    ll m[maxn][maxn];
}ans,res;
ll n,N;

matrix mul(matrix a,matrix b){
    matrix tmp;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            tmp.m[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            for(int k=1;k<=n;k++){
                tmp.m[i][j]=(tmp.m[i][j]+(a.m[i][k]*b.m[k][j])%mod)%mod;
            }
        }
    }
    return tmp;
}
void quickpow(ll N,ll n){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i==j)ans.m[i][j]=1;
            else ans.m[i][j]=0;
        }
    }
    while(N){
        if(N&1)ans=mul(res,ans);
        res=mul(res,res);
        N=N>>1;
    }
}
int main(){
    scanf("%lld%lld",&n,&N);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            scanf("%lld",&res.m[i][j]);
        }
    }
    quickpow(N,n);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<ans.m[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}

 

矩阵快速幂模板

标签:整数   end   代码   html   lse   col   com   i++   space   

原文地址:https://www.cnblogs.com/mohari/p/12945187.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!