Tr A HDU - 1575（矩阵快速幂）

题目描述

A为一个方阵，则Tr A表示A的迹（就是主对角线上各项的和），现要求Tr(A^k)%9973。

输入

数据的第一行是一个T，表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行，每行有n个数据，每个数据的范围是[0,9]，表示方阵A的内容。

输出

对应每组数据，输出Tr(A^k)%9973。

样例输入

2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9

样例输出

2
2686

---

思路

直接矩阵快速幂

---

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <math.h>
#define mem(a) memset(a,0,sizeof(a))
using namespace std;

int mod;
int n;
struct matrix
{
    int a[12][12];
    int r;
    int c;
};

matrix multiply (matrix x,matrix y)
{
    matrix t;
    t.r=x.r;
    t.c=y.c;
    for(int i=0; i<x.r; ++i)
        for(int j=0; j<y.c; ++j)
            t.a[i][j]=0;
    for(int i=0; i<x.r; ++i)
        for(int j=0; j<y.c; ++j)
            for(int k=0; k<x.r; ++k)
            {
                t.a[i][j]+=x.a[i][k]*y.a[k][j];
                t.a[i][j]%=mod;
            }
    return t;
}

matrix powmatrix (matrix x,int y)
{
    matrix t=x;
    for(int i=0; i<n; ++i)
        for(int j=0; j<n; ++j)
            if(i==j)
                t.a[i][j]=1;
            else
                t.a[i][j]=0;
    while(y)
    {
        if(y&1)
            t=multiply(t,x);
        y/=2;
        x=multiply(x,x);
    }
    return t;
}

int main ()
{
#ifndef ONLINE_JUDGE
    freopen("1","r",stdin);
#endif // ONLINE_JUDGE
    ios_base::sync_with_stdio(false);
    cin.tie(0);

    int k,T;
    matrix t,x;
    int ans;
    cin>>T;
    mod = 9973;
    while(T--)
    {
        cin>>n>>k;
        x.c=n;
        x.r=n;
        for(int i=0; i<n; ++i)
            for(int j=0; j<n; ++j)
                cin>>x.a[i][j];
        t=powmatrix(x,k);
        ans=0;
        for(int i=0; i<n; ++i)
            ans+=t.a[i][i],ans%=mod;
        cout<<ans<<endl;
    }
    return 0;
}