/*愧疚了,好久没写什么有意思的题了。
这道基于二分的矩阵连乘,方法很经典。
求幂二分,求和再二分。经典。
求幂的时候,若奇数,A^n=A^(n/2)*A^(n/2)*A.
偶数时 A^n=A^(n/2)*A^(n/2)。
求和的时候,若为奇数,A^1+A^2+...+A^(2*m+1)=(A^1+...+A^m)+(A^1+...+A^m)*A^m*A,
若为偶数 A^1+A^2+...+A^(2*m+1)=(A^1+...+A^m)+(A^1+...+A^m)*A^m。
奇数的时候,构造0,1单位矩阵。
对奇数进行特判。
memcpy(a,b,sizeof(int)*N*N);
切记memcpy的赋值范围。之前因为太小,答案一直错误。*/
#include <stdio.h>
#include <cstring>
int n,m,k;
const int N=31;
int ans[N][N],A[N][N];
void add(int a[][N],int b[][N])
{
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
{
a[i][j]+=b[i][j];
if(a[i][j]>=m) a[i][j]%=m;
}
}
void mul(int a[][N],int b[][N])
{
int tmp[N][N];
for(int i=0; i<n; i++)
{
for(int j=0; j<n; j++)
{
tmp[i][j]=0;
for(int k=0; k<n; k++)
{
tmp[i][j]+=a[i][k]*b[k][j];
if(tmp[i][j]>=m) tmp[i][j]%=m;
}
}
}
memcpy(a,tmp,sizeof(tmp));
}
void pow(int a[][N],int b[][N],int x)
{
int tmp[N][N];
memcpy(tmp,b,sizeof(int)*N*N);
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
{
if(i==j) a[i][j]=1;
else a[i][j]=0;
}
while(x)
{
if(x&1)//最后的结果存在数组a中,因为最后x比为1,符合条件,自己模拟一遍则易知.
mul(a,tmp);
mul(tmp,tmp);
x>>=1;
}
}
void mat_sum(int a[][N],int b[][N],int x)
{
int c[N][N],d[N][N];
if(x==0)
{
memset(a,0,sizeof(int)*N*N);
return ;
}
if(x==1)
{
memcpy(a,b,sizeof(int)*N*N);
return ;
}
mat_sum(a,b,x>>1);
pow(c,b,x>>1);
memcpy(d,a,sizeof(int)*N*N);
mul(a,c);
add(a,d);
if(x&1)
{
mul(c,c);
mul(c,b);
add(a,c);
}
}
int main()
{
while(scanf("%d%d%d",&n,&k,&m)==3)
{
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
scanf("%d",&A[i][j]);
mat_sum(ans,A,k);
for(int i=0; i<n; i++)
{
printf("%d",ans[i][0]);
for(int j=1; j<n; j++)
{
if(j<n-1) printf(" %d",ans[i][j]);
else printf(" %d\n",ans[i][j]);
}
}
}
return 0;
}
poj 3233 Matrix Power Series 矩阵快速幂
最新推荐文章于 2019-08-30 15:09:12 发布