HDU 1575 Tr A

Tr A
Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4370 Accepted Submission(s): 3274
Problem Description
A为一个方阵，则Tr A表示A的迹（就是主对角线上各项的和），现要求Tr(A^k)%9973。
Input
数据的第一行是一个T，表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行，每行有n个数据，每个数据的范围是[0,9]，表示方阵A的内容。
Output
对应每组数据，输出Tr(A^k)%9973。
Sample Input
2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9
Sample Output
2
2686
Author
xhd
Source
HDU 2007-1 Programming Contest

/*
矩阵乘法快速幂.
A矩阵"自乘"k-1次后求主对角线和.
用快速幂跑(非递归版).
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define MAXN 11
#define p 9973
using namespace std;
int n,K,d[MAXN][MAXN],b[MAXN][MAXN],tot,c[MAXN][MAXN],ans[MAXN][MAXN],total;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x*f; 
}
void mi()
{
    while(K)
    {
        if(K&1)
        {
            for(int i=1;i<=n;i++)
              for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                  c[i][j]=(c[i][j]+(ans[i][k]*b[k][j])%p)%p;
            for(int i=1;i<=n;i++)
              for(int j=1;j<=n;j++)
                ans[i][j]=c[i][j],c[i][j]=0;
        }
        for(int i=1;i<=n;i++)
          for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
              c[i][j]=(c[i][j]+(b[i][k]*b[k][j])%p)%p;
        for(int i=1;i<=n;i++)
          for(int j=1;j<=n;j++)
            b[i][j]=c[i][j],c[i][j]=0;

        K>>=1;
    }
}
void slove()
{
    K--;total=0;
    mi();
    for(int i=1;i<=n;i++) 
      for(int j=1;j<=n;j++)
        if(i==j) total=(total+ans[i][j])%p;
}
int main()
{
    int t;
    t=read();
    while(t--)
    {
        n=read();K=read();
        for(int i=1;i<=n;i++)
          for(int j=1;j<=n;j++)
            b[i][j]=read(),ans[i][j]=b[i][j];
        slove();
        printf("%d\n",total);
    }
    return 0;
}