题意:矩阵求和
分析:如果用矩阵快速幂直接把每一项求出来相加会超时。可利用公式,使用二分
快速求出
(1)当时,
(2)当时,那么有
(3)当时,那么有
代码:
#include <iostream>
#include <cstdio>
using namespace std;
struct Matrax
{
int mat[100][100];
}U,F;
int N,MOD,K;
Matrax add(Matrax a,Matrax b)
{
Matrax ans;
for(int i=0;i<N;i++)
for(int j=0;j<N;j++)
ans.mat[i][j]=(a.mat[i][j]%MOD+b.mat[i][j]%MOD)%MOD;
return ans;
}
Matrax multi(Matrax a,Matrax b)
{
Matrax ans;
for(int i=0;i<N;i++)
{
for(int j=0;j<N;j++)
{
ans.mat[i][j]=0;
for(int k=0;k<N;k++)
ans.mat[i][j]+=(a.mat[i][k]*b.mat[k][j])%MOD;
ans.mat[i][j]%=MOD;
}
}
return ans;
}
Matrax mkpow(int n)
{
Matrax ans=U,p=F;
while(n)
{
if(n&1)
ans=multi(ans,p);
p=multi(p,p);
n/=2;
}
return ans;
}
Matrax solve(int k)
{
if(k==1)
return F;
Matrax ans=solve(k/2);
if(k&1)
{
Matrax c=multi(ans,mkpow(k/2+1));
c=add(c,mkpow(k/2+1));
ans=add(ans,c);
}
else
{
Matrax c=multi(ans,mkpow(k/2));
ans=add(ans,c);
}
return ans;
}
int main()
{
int i,j;
while(cin>>N>>K>>MOD)
{
for(i=0;i<N;i++)
for(j=0;j<N;j++)
{
scanf("%d",&F.mat[i][j]);
F.mat[i][j]%=MOD;
U.mat[i][j]=(i==j);
}
Matrax ans=solve(K);
for(i=0;i<N;i++)
{
for(j=0;j<N;j++)
printf("%d ",ans.mat[i][j]);
printf("\n");
}
}
return 0;
}