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