求等比矩阵和:S = A + A2 + A3 + … + Ak
经典算法:
构造矩阵:
B= A E
0 E
其中E为单位矩阵,和A同阶
然后矩阵快速幂计算B^(K+1)
即:
B^(k+1)= A^(k+1) E+A^1+A^2+......+A^K
0 E
PS:这NM不让我用tab么
也就是说这样求出来的矩阵的右上角的子矩阵减去一个单位矩阵E便是要求的S
#include<stdio.h>
#include<string.h>
#define MAXN 70
#define ll long long
struct matrix
{
ll m[MAXN][MAXN];
};
int n;
ll m;
matrix multi(const matrix &a,const matrix &b)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j];
ans.m[i][j]%=m;
}
return ans;
}
matrix pow(matrix a,int k)
{
matrix ans;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(i==j)ans.m[i][j]=1;
else ans.m[i][j]=0;
while(k)
{
if(k&1) ans=multi(ans,a);
a=multi(a,a);
k>>=1;
}
return ans;
}
int main()
{
matrix a,ans;
int k;
while(scanf("%d%d%lld",&n,&k,&m)!=EOF)
{
memset(&ans,0,sizeof(ans));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
scanf("%lld",&a.m[i][j]);
a.m[i][j]%=m;
}
for(int i=n;i<(n<<1);i++)
a.m[i-n][i]=a.m[i][i]=1;
n=n<<1;
ans=pow(a,k+1);
n>>=1;
for(int i=0;i<n;i++)
ans.m[i][i+n]=(ans.m[i][i+n]-1+m)%m;
for(int i=0;i<n;i++)
{
printf("%lld",ans.m[i][n]);
for(int j=1;j<n;j++)
printf(" %lld",ans.m[i][j+n]);
printf("\n");
}
}
return 0;
}