la 3704 - Cellular Automaton(矩阵优化递推)

la 3704 - Cellular Automaton

循环矩阵，注意用O(n^2)的复杂度优化乘法

还有就是不要在函数里定义矩阵，会爆栈，定义成全局

#include<stdio.h>
#include<string.h>
#define MAXN 505
#define ll  long long
struct mat
{
	ll m[MAXN][MAXN];
};
int n,mod;
mat mul_mat,pow_mat;
void multi(const mat &a,const mat &b)
{
	for(int i=0;i<n;i++)
	{
		mul_mat.m[0][i]=0;
		for(int j=0;j<n;j++)
			mul_mat.m[0][i]+=a.m[0][j]*b.m[j][i];
		mul_mat.m[0][i]%=mod;
	}
	for(int i=1;i<n;i++)
		for(int j=0;j<n;j++)
			mul_mat.m[i][j]=mul_mat.m[i-1][(j+n-1)%n];
}
void pow(mat &a,int k)
{
	memset(&pow_mat,0,sizeof(pow_mat));
	for(int i=0;i<n;i++) pow_mat.m[i][i]=1;
	while(k)
	{
		if(k&1)
		{
			multi(pow_mat,a);
			pow_mat=mul_mat;
		}
		multi(a,a);
		a=mul_mat;
		k>>=1;
	}
}
mat a;
int b[MAXN];
int main()
{
	int d,k;
	while(scanf("%d%d%d%d",&n,&mod,&d,&k)!=EOF)
	{
		for(int i=0;i<n;i++) scanf("%d",&b[i]);
		memset(&a,0,sizeof(a));
		for(int i=0;i<n;i++)
			for(int j=i-d;j<=i+d;j++)
				a.m[i][(j+n)%n]=1;
		pow(a,k);
		ll ans;
		for(int i=0;i<n;i++)
		{
			ans=0;
			for(int j=0;j<n;j++)
			{
				ans+=pow_mat.m[i][j]*b[j];
				ans%=mod;
			}
			if(i) printf(" ");
			printf("%lld",ans);
		}
		printf("\n");
	}
	return 0;
}