题目就是求一个矩阵的1~k次幂相加
经典题目
代码如下
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
struct mat
{
int x[100][100];
}A,B,E;
int n,k,mod;
mat operator *(mat a,mat b)
{
mat c;
for(int i=0;i<2*n;i++)
{
for(int j=0;j<2*n;j++)
{
c.x[i][j]=0;
for(int k=0;k<2*n;k++)
{
c.x[i][j]+=(a.x[i][k]*b.x[k][j]);
c.x[i][j]%=mod;
}
}
}
return c;
}
mat operator ^(mat x,int y)
{
mat b=x,r=E;
while(y)
{
if(y&1)
r=r*b;
b=b*b;
y>>=1;
}
return r;
}
int main()
{
// freopen("kepin.txt","r",stdin);
// freopen("output.txt","w",stdout);
while(~scanf("%d%d%d",&n,&k,&mod))
{
memset(A.x,0,sizeof(A.x));
memset(B.x,0,sizeof(B.x));
memset(E.x,0,sizeof(E.x));
for(int i=0;i<2*n;i++)
{
for(int j=0;j<2*n;j++)
{
if(i==j)
E.x[i][j]=1;
}
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&A.x[i][j]);
}
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
B.x[i][j]=A.x[i][j];
}
B.x[n+i][i]=B.x[n+i][n+i]=1;
}
B=B^(k+1);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
int a=B.x[i+n][j]%mod;
if(i==j)
a=(a+mod-1)%mod;
printf("%d%c",a,j==n-1?'\n':' ');
}
}
}
return 0;
}