1113 矩陣快速冪

給出一個N * N的矩陣,其中的元素均爲正整數。求這個矩陣的M次方。由於M次方的計算結果太大,只需要輸出每個元素Mod (10^9 + 7)的結果。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100;
const long long mod=1000000007;
typedef long long ll;
struct Mat
{
	ll a[N][N];
};
int n;
Mat mitriplus(Mat A,Mat B,int m)
{
	Mat ans;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			ans.a[i][j]=0;
			for(int k=0;k<n;k++)
			ans.a[i][j]+=(A.a[i][k]*B.a[k][j])%m;
			ans.a[i][j]%=m;
		}
	}
	return ans;
}
Mat qpow(Mat M,ll n,ll m)
{
	Mat ans=M;
	while(n)
	{
		if(n&1)
		ans=mitriplus(ans,M,m);
		M=mitriplus(M,M,m);
		n>>=1;
	}
	return ans;
}
int main()
{
	ll m;
	while(scanf("%d %I64dd",&n,&m)!=EOF)
	{
		Mat A;
		for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
		scanf("%I64d",&A.a[i][j]);
		Mat ans=qpow(A,m-1,mod);
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n-1;j++)
			printf("%I64d ",ans.a[i][j]);
			printf("%I64d\n",ans.a[i][n-1]);
		}
	}
	return 0;
}
		
	


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章