矩阵快速幂模板

#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
#include<math.h>
#include<cstdio>
#include<sstream>
#include<numeric>//STL数值算法头文件
#include<stdlib.h>
#include <ctype.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<functional>//模板类头文件
using namespace std;

typedef long long ll;
const int maxn=20;
const int INF=0x3f3f3f3f;

const int mod=9973;

int n,m;
struct mat
{
    int arr[maxn][maxn];
} str;

mat mul(mat& A,mat& B)
{
    mat res;
    memset(res.arr,0,sizeof(res.arr));
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            for(int k=0; k<n; k++)
                res.arr[i][j]=((res.arr[i][j]+A.arr[i][k]*B.arr[k][j])%mod+mod)%mod;
    return res;
}

mat pow(mat &A,int k)
{
    mat M;
    memset(M.arr,0,sizeof(M.arr));
    for(int i=0; i<n; i++)
        M.arr[i][i]=1;
    while(k)
    {
        if(k&1) M=mul(M,A);
        A=mul(A,A);
        k>>=1;
    }
    return M;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&m);
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++)
                scanf("%d",&str.arr[i][j]);
        int sum=0;
        str=pow(str,m);
        for(int i=0; i<n; i++)
            sum+=str.arr[i][i];
        printf("%d\n",sum%mod);
    }
    return 0;
}
posted @ 2017-05-30 17:06  xushukui  阅读(179)  评论(0编辑  收藏  举报