HDU 1575 Tr A （矩阵乘法）

Tr A

Time Limit : 1000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 11   Accepted Submission(s) : 7

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

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

Author

xhd

Source

HDU 2007-1 Programming Contest

矩阵乘法与快速幂模板

设A为矩阵,则A^k可以化为A^(k/2)*A^(k/2);k为偶数，A^(k/2)*A^(k/2)*A，k为奇数；

#include<stdio.h>
#include<string.h>
struct node
{
  __int64 mat[15][15];
};//矩阵型结构体
__int64 n;
node mat_mat(node a,node b)//矩阵乘法
{
  node c;
  memset(c.mat,0,sizeof(c.mat));
  for(int i=1;i<=n;i++)
       {
               for(int j=1;j<=n;j++)
               {
                       for(int k=1;k<=n;k++)
                       {
                               c.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
                               c.mat[i][j]%=9973;
                       }
               }
       }
  return c;
}     
node pow(node a,__int64 nn)
{
   node aa;
   memset(aa.mat,0,sizeof(aa.mat));
   for(int i=1;i<=n;i++)
   aa.mat[i][i]=1;
   while(nn)
   {
     if(nn&1)
     aa=mat_mat(a,aa);
     a=mat_mat(a,a);
     nn>>=1;
   }
   return aa;
}
int main ()
{
  node a,b;
  __int64 i,j,sum,nn,t;
  scanf("%I64d",&t);
  while(t--)
  {
     scanf("%I64d%I64d",&n,&nn);
     for(i=1;i<=n;i++)
     for(j=1;j<=n;j++)
     {
        scanf("%I64d",&a.mat[i][j]);
        a.mat[i][j]%=9973;
     }
     b=pow(a,nn);
     for(sum=0,i=1;i<=n;i++)
     {
       sum+=b.mat[i][i];
       sum%=9973;
     }
     printf("%I64d\n",sum);
   }
   return 0;
}