hdu1575 Tr A

Posted 2020-10-12 ---学习ing---

 

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

一面赶期中考，一面刷水题，QwQ，感觉第一次大学考试药丸辣。。。

#include <bits/stdc++.h>
using namespace std;
#define LL long long 
const int maxn=12;
int n;
struct mat
{
     int  m[maxn][maxn]; 
     mat(){memset(m,0,sizeof(m));};
     mat friend operator*(mat a,mat b){
         mat d;
         for(int i=0;i<n;i++)
          for(int j=0;j<n;j++)
           for(int k=0;k<n;k++)
            d.m[i][j]=(d.m[i][j]+a.m[i][k]*b.m[k][j]%9973)%9973;
        return d;     
     }    
};
mat pow(mat x,int cnt){
    mat d;
    for(int i=0;i<n;i++) d.m[i][i]=1;
    while(cnt){
        if(cnt&1) d=d*x;
        x=x*x;
        cnt>>=1;
    }
    return d;
}
int main()
{
    int i,j,k,num,T;
    scanf("%d",&T);
    while(T--){
        mat ans;
        scanf("%d%d",&n,&k);
        num=0;
        for(i=0;i<n;i++)
         for(j=0;j<n;j++)
         scanf("%d",&ans.m[i][j]);
        ans=pow(ans,k);
        for(i=0;i<n;i++) num+=ans.m[i][i];
        printf("%d\n",num%9973);
    }
    return 0;
}

View Code