洛谷P3390 【模板】矩阵快速幂

注意点：1.矩阵运算规则：题解里讲的很详细。

2.这个题只能用while型的快速幂，不能用递归型的，会TLE。

3.极其坑爹的一点：常量定义居然不能用define，只能用const？？判定非常迷，卡了我3个小时！

上AC代码

#include<bits/stdc++.h>
//#define mod 100000007     //这个不行，会MLE！！用下面的常量定义 
using namespace std;
const int mod=1000000007;

struct matrix{
    long long m[105][105];
}A;

long long n,k;

matrix operator *(const matrix &a,const matrix &b){
    matrix z;
    for(int i=1;i<=n;i++)     
        for(int j=1;j<=n;j++)
          z.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++){
                z.m[i][j]+=(a.m[i][k]*b.m[k][j]%mod);
                z.m[i][j]%=mod;
            }
        }
    }
    return z;
}

int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>A.m[i][j];
        }
    }
    matrix ans;
    for(int i=1;i<=n;i++) ans.m[i][i]=1;
    while(k>0){
        if(k%2==1) ans=ans*A;
        A=A*A;
        k/=2;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++) cout<<ans.m[i][j]<<' ';
        cout<<endl;
    }
    return 0;
}