HDOJ1575 Tr A ----- 矩阵快速幂模板题

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

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 <algorithm>
#include <cstring>
using namespace std;
const int mod = 9973;
int n,k;
// 定义方阵
struct Matrix {
    int data[15][15];
    Matrix() {memset(data,0,sizeof(data));}
    friend Matrix operator * (const Matrix &a,const Matrix &b) {
        Matrix t;
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++) 
                for(int m = 1;m <= n;m++) {
                    t.data[i][j] = (t.data[i][j] + a.data[i][m]*b.data[m][j]) % mod;
                }
        return t;        
    }
}mat;


Matrix Multi(Matrix mat,int k) {
    if(k == 1)
        return mat;
    int mid = k / 2;
    Matrix ret = Multi(mat,mid);
    if(k & 1) {
        // k是奇数
        return ret * ret * mat;
    } else {
        return ret * ret;
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--) {
        scanf("%d%d",&n,&k);
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++) {
                scanf("%d",&mat.data[i][j]);
            }
        Matrix ans = Multi(mat,k);
        int result = 0;
        for(int i = 1;i <= n;i++)
            result += ans.data[i][i];
        printf("%d\n",result%mod);
    }
    
    return 0;
}