B - Tr A HDU - 1575

A为一个方阵,则Tr A表示A的迹(就是主对角线上各项的和),现要求Tr(A^k)%9973。 
Input
数据的第一行是一个T,表示有T组数据。 
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行,每行有n个数据,每个数据的范围是 0,9 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

矩阵的快速幂运算要注意和记住的以下几点:

1:矩阵相乘函数中这里最重要:temp[i][j]=(temp[i][j]+ak[i][k]*bk[k][j])%mod;

2:快速幂函数中单位矩阵的初始化

就以上两点值得注意吧。

#include<bits/stdc++.h>
#define mod 9973
using namespace std;
int T,n,k;
int a[11][11];
void multi(int ak[][11],int bk[][11])
{
	int temp[11][11];
	memset(temp,0,sizeof(temp));
	int i,j,k;
	
	for(i=0;i<n;i++)
	for(j=0;j<n;j++)
	for(k=0;k<n;k++)
	{
		temp[i][j]=(temp[i][j]+ak[i][k]*bk[k][j])%mod;
	}
	
	for(i=0;i<n;i++)
	for(j=0;j<n;j++)
	ak[i][j]=temp[i][j];
}

int quick(int ak[][11])
{
	int i,j;
	int E[11][11];
	memset(E,0,sizeof(E));
	for(i=0;i<11;i++)
	E[i][i]=1;
	
	while(k>0)
	{
		if(k&1)
		{
			multi(E,ak);
		}
		k>>=1;
		multi(ak,ak);
	}
	
	for(i=0;i<n;i++)
	for(j=0;j<n;j++)
	ak[i][j]=E[i][j];
}
int main()
{
	int i,j;
	cin>>T;
	while(T--)
	{
		cin>>n>>k;
		for(i=0;i<n;i++)
		for(j=0;j<n;j++)
		cin>>a[i][j];
		
		quick(a);
		int ans=0;
		for(i=0;i<n;i++)
		ans=(ans+a[i][i])%mod;
		
		printf("%d\n",ans);
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值