hdoj 1575 Tr A

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1575

解题思路：矩阵快速幂

#include<iostream>
using namespace std;

int n;  //nxn矩阵

struct Matrix{
    int a[10][10];
}origin,res;

Matrix multiply(Matrix x,Matrix y){
    Matrix temp;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            temp.a[i][j]=0;
            for(int k=0;k<n;k++) temp.a[i][j]+=x.a[i][k]*y.a[k][j];
            temp.a[i][j]%=9973;
        }
    }
    return temp;
}

void calc(int k){
    while(k){
        if(k&1) res=multiply(res,origin);
        k>>=1;
        origin=multiply(origin,origin);
    }
}

int main(){
    int T,k,i,j;
    cin>>T;
    while(T--){
        cin>>n>>k;

        //初始化origin和res(单位矩阵)
        for(i=0;i<n;i++){
            for(j=0;j<n;j++){
                cin>>origin.a[i][j];
                res.a[i][j]=0;
            }
            res.a[i][i]=1;
        }

        calc(k);
        int ans=0;
        for(i=0;i<n;i++) ans+=res.a[i][i];
        ans%=9973;
        cout<<ans<<endl;
    }
    return 0;
}

---