HDU1575:Tr A(矩阵快速幂模板题）

http://acm.hdu.edu.cn/showproblem.php?pid=1575

 

#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <cstdio>
#include <algorithm>
#define mod 9973
using namespace std;
struct matrix
{
    int a[][];
} init,res;
int n,k;
matrix Mult(matrix x,matrix y)
{
    matrix tmp;
    for(int i=; i<n; i++)
    {
        for(int j=; j<n; j++)
        {
            tmp.a[i][j]=;
            for(int k=; k<n; k++)
                tmp.a[i][j]=(tmp.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
        }
    }
    return tmp;
}
matrix Pow(matrix x,int k)
{
    matrix tmp;
    for(int i=; i<n; i++)
    {
        for(int j=; j<n; j++)
            tmp.a[i][j]=(i==j);
    }
    while(k)
    {
        if(k&)
            tmp=Mult(tmp,x);
        k>>=;
        x=Mult(x,x);
    }
    return tmp;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>k;
        for(int i=; i<n; i++)
        {
            for(int j=; j<n; j++)
                scanf("%d",&init.a[i][j]);
        }
        res=Pow(init,k);
        int sum=;
        for(int i=; i<n; i++)
            sum=(sum+res.a[i][i])%mod;
        cout<<sum<<endl;
    }
    return ;
}