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;
}
		
	


posted @ 2016-03-05 18:49  __NaCl  阅读(160)  评论(0编辑  收藏  举报