矩阵快速幂 裸 hdu1575

裸题，求A^n次后的对角线数字之和

 #include<cstdio>
 #include<algorithm>
 #include<string.h>
 using namespace std;
 const int mod=;
 int n,m;
 struct node
 {
     int a[][];
 }ans,base;
 node mat(node x,node y)
 {
     node c;
     for(int i=;i<n;i++)
     for(int j=;j<n;j++){
         c.a[i][j]=;
     }
     for(int i=;i<n;i++)
     for(int j=;j<n;j++)
     for(int k=;k<n;k++){
         c.a[i][j]=(c.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
     }
     return c;
 }
 node quick_mod()
 {
     node tmp;
     for(int i=;i<=n;i++)
     for(int j=;j<=n;j++){
         if(i==j) tmp.a[i][j]=;
         else tmp.a[i][j]=;
     }
     while(m){
         if(m&) tmp=mat(tmp,base);
         base=mat(base,base);
         m>>=;
     }
     return tmp;
 }
 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--){
         scanf("%d%d",&n,&m);
         for(int i=;i<n;i++)
         for(int j=;j<n;j++){
             scanf("%d",&base.a[i][j]);
         }
         ans=quick_mod();
         int tmp=;
         for(int i=;i<n;i++)
             tmp=(tmp+ans.a[i][i])%mod;
         printf("%d\n",tmp);
     }
     return ;
 }