HDU 1575 Tr A

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

裸裸的矩阵快速幂，今天复习了一下，直接模板

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define mod 9973
using namespace std;
struct node {
    int map[11][11];
};
node a;
int t,n,k;
inline void read(int&x) {
    x=0;char c=getchar();
    while(c>'9'||c<'0') c=getchar();
    while(c>='0'&&c<='9') x=10*x+c-48,c=getchar();
}
inline node mul(node x,node y) {
    node z;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++) {
        z.map[i][j]=0;
        for(int l=1;l<=n;l++) 
          z.map[i][j]+=x.map[i][l]*y.map[l][j];
          z.map[i][j]%=mod;
      }
    return z;
}
inline void jzc(int t) {
    node b;
    for(int i=1;i<=n;i++) 
      for(int j=1;j<=n;j++)
        b.map[i][j]=(i==j);
    while(t) {
        if(t&1) b=mul(b,a);
        a=mul(a,a);
        t>>=1;
    }
    int ans=0;
    for(int i=1;i<=n;i++) ans+=b.map[i][i],ans%=mod;
    printf("%d\n",ans);
}
int hh() {
    read(t);
    while(t--) {
        read(n);read(k);
        for(int i=1;i<=n;i++)
          for(int j=1;j<=n;j++)
            read(a.map[i][j]);
        jzc(k);
    }
    return 0;
}
int hhh=hh();
int main() {;}