H - Tr A

点击打开链接

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<cstring>
int a[1005][1005],b[1005][1005],c[1005][1005],d[1005][1005];
int main()
{
	int T,n;
	long long t;
	int i,j,k;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d %lld",&n,&t);
		for(i=1;i<=n;i++)
		  for(j=1;j<=n;j++)
		  {
		  	scanf("%d",&c[i][j]);
		    a[i][j]=b[i][j]=c[i][j];
		  }
		t=t-1;
		while(t!=0)
		{
			if(t&1)
			{
			  for(i=1;i<=n;i++)
		        for(j=1;j<=n;j++)	
			      d[i][j]=c[i][j];
			  memset(c,0,sizeof(c));
			  for(i=1;i<=n;i++)
		        for(j=1;j<=n;j++)
		        {
		  	     k=0;
		         while(k<n)
		  	     {
		  	      k++;
			      c[i][j]=(c[i][j]+a[i][k]*d[k][j])%9973;	
			     }
		        }		
			}
			t/=2;
			memset(d,0,sizeof(d));
			for(i=1;i<=n;i++)
		      for(j=1;j<=n;j++)
		      {
		  	    k=0;
		        while(k<n)
		  	    {
		  	      k++;
			      d[i][j]=(d[i][j]+a[i][k]*b[k][j])%9973;	
			    }
		      }	
		    for(i=1;i<=n;i++)
		      for(j=1;j<=n;j++)
			    a[i][j]=b[i][j]=d[i][j];	 
		}
		int sum=0;
		for(i=1;i<=n;i++)
		  sum+=c[i][i];
		sum%=9973;
		printf("%d\n",sum); 
	}
}