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

矩阵快速幂模板

时间:2020-07-18 22:14:26      阅读:75      评论:0      收藏:0      [点我收藏+]

标签:scan   long   operator   style   ++   turn   ace   struct   模板   

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=105;
const int mod=1e9+7;
int n;
struct matrix {
    ll a[maxn][maxn];
    matrix() {
        memset(a,0,sizeof(a));
    }
    void build () {
        for (int i=1;i<=n;i++) a[i][i]=1;
    }
}wjm;
matrix operator * (const matrix &x,const matrix &y) {
    matrix ans;
    for (int k=1;k<=n;k++)
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                ans.a[i][j]=(ans.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
    return ans;
}
int main () {
    ll k;
    scanf("%d%lld",&n,&k);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            scanf("%lld",&wjm.a[i][j]);
    matrix ans;
    ans.build();
    do {
        if (k&1) ans=ans*wjm;
        wjm=wjm*wjm;
        k>>=1;
    }while (k);
    for (int i=1;i<=n;i++) {
        for (int j=1;j<=n;j++) printf("%d ",ans.a[i][j]);
        printf("\n");
    }
}

 

矩阵快速幂模板

标签:scan   long   operator   style   ++   turn   ace   struct   模板   

原文地址:https://www.cnblogs.com/zhanglichen/p/13337459.html

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