矩阵快速幂 裸 hdu1575

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

 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<string.h>
 4 using namespace std;
 5 const int mod=9973;
 6 int n,m;
 7 struct node
 8 {
 9     int a[15][15];
10 }ans,base;
11 node mat(node x,node y)
12 {
13     node c;
14     for(int i=0;i<n;i++)
15     for(int j=0;j<n;j++){
16         c.a[i][j]=0;
17     }
18     for(int i=0;i<n;i++)
19     for(int j=0;j<n;j++)
20     for(int k=0;k<n;k++){
21         c.a[i][j]=(c.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
22     }
23     return c;
24 }
25 node quick_mod()
26 {
27     node tmp;
28     for(int i=0;i<=n;i++)
29     for(int j=0;j<=n;j++){
30         if(i==j) tmp.a[i][j]=1;
31         else tmp.a[i][j]=0;
32     }
33     while(m){
34         if(m&1) tmp=mat(tmp,base);
35         base=mat(base,base);
36         m>>=1;
37     }
38     return tmp;
39 }
40 int main()
41 {
42     int T;
43     scanf("%d",&T);
44     while(T--){
45         scanf("%d%d",&n,&m);
46         for(int i=0;i<n;i++)
47         for(int j=0;j<n;j++){
48             scanf("%d",&base.a[i][j]);
49         }
50         ans=quick_mod();
51         int tmp=0;
52         for(int i=0;i<n;i++)
53             tmp=(tmp+ans.a[i][i])%mod;
54         printf("%d\n",tmp);
55     }
56     return 0;
57 }