HDU1575矩阵快速幂裸题

Tr A

                                    Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
                                                    Total Submission(s): 2013    Accepted Submission(s): 1490

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 <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
 

using namespace std;
 

#define maxn 12
 

struct Matrix
{
    int a[maxn][maxn];
}res,org;
 

int n,k;
 

Matrix multiply(Matrix A,Matrix B)
{
    Matrix C;
    memset(C.a,0,sizeof(C.a));
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            for(int k=1;k<=n;k++)
            {
                C.a[i][j]+=A.a[i][k]*B.a[k][j];
                C.a[i][j]%=9973;
            }
        }
    }
    return C;
}
 

int quick_pow(Matrix res,int k)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            res.a[i][j]=(i==j);//构造单位矩阵；
        }
    }
    while(k)
    {
        if(k&1)
        {
            res=multiply(res,org);
        }
        org=multiply(org,org);
        k>>=1;
    }
    int sum=0;
    for(int i=1;i<=n;i++)
    {
            sum+=res.a[i][i];
            sum%=9973;
    }
    return sum;
}
 

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>k;

        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                cin>>org.a[i][j];
            }
        }
        int ans=quick_pow(org,k);
        cout<<ans<<endl;
    }
    return 0;
}
 

代码风格有点挫，大神勿喷。