hdu 1575 Tr A

矩阵裸体

#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <iostream>
using namespace std;
#define MOD 9973
#define MN 10
#define MM 10
int N;
struct matrix
{
       int a[MN][MM];
}origin,res,ans;
matrix multiply(matrix &x,matrix &y)
{
       matrix temp;
   //    memset(temp.a,0,sizeof(temp.a));
       for(int i=0;i<N;i++)
       {
           for(int j=0;j<N;j++)
            {
                temp.a[i][j]=0;
                for(int k=0;k<N;k++)
                {
                    temp.a[i][j]=(temp.a[i][j]+(x.a[i][k])*(y.a[k][j]));
                }
                temp.a[i][j]%=MOD;
            }
       }
       return temp;
}
void init()
{
     for(int i=0;i<N;i++)
     {
         for(int j=0;j<N;j++)
         {
             scanf("%d",&origin.a[i][j]);
         }
     }
     memset(res.a,0,sizeof(res.a));
     for(int i=0;i<N;i++)
        res.a[i][i]=1;
}

void calc(int n)
{
     while(n!=1)
     {
        if(n&1){n--;
        res=multiply(res,origin);}
        n>>=1;
        origin=multiply(origin,origin);
     }
     ans=multiply(res,origin);
}
int main()
{
    int k,cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d %d",&N,&k);
        init();
        calc(k);
        int sum=0;
        for(int i=0;i<N;i++)
            sum=(sum+ans.a[i][i])%MOD;
        printf("%d\n",sum);
    }
    return 0;
}