【矩阵乘法】Gym - 101412C - One-Dimensional Cellular Automaton

给你一个一维细胞自动机，第i个格子在时刻t的状态是这样获得的，问你t时刻的状态。

把0时刻的状态视作一个列向量，发现状态转移其实是一个n*n的矩阵（以n=5为例），

B	C
A	B	C
	A	B	C
		A	B	C
			A	B

 

直接快速幂即可。

#include<cstdio>
#include<vector>
using namespace std;
typedef vector<int> vec;
typedef vector<vec> mat;
int n,m,A,B,C,T;
mat operator * (const mat &A,const mat &B){
	mat C(A.size(),vec(B[0].size()));
	for(int i=0;i<A.size();++i){
		for(int k=0;k<B.size();++k){
			for(int j=0;j<B[0].size();++j){
				C[i][j]=(C[i][j]+A[i][k]*B[k][j])%m;
			}
		}
	}
	return C;
}
mat I;
mat Quick_Pow(mat a,int p){
	if(!p){
		return I;
	}
	mat res=Quick_Pow(a,p>>1);
	res=res*res;
	if(p&1){
		res=res*a;
	}
	return res;
}
int main(){
//	freopen("c.in","r",stdin);
	while(1){
		scanf("%d%d%d%d%d%d",&n,&m,&A,&B,&C,&T);
		if(!n && !m && !A && !B && !C && !T){
			return 0;
		}
		mat P(n,vec(1));
		int x;
		for(int i=0;i<n;++i){
			scanf("%d",&x);
			P[i][0]=x;
		}
		I.assign(n,vec(n));
		for(int i=0;i<n;++i){
			for(int j=0;j<n;++j){
				I[i][j]=(i==j);
			}
		}
		mat R(n,vec(n));
		for(int i=0;i<n;++i){
			if(i-1>=0){
				R[i][i-1]=A;
			}
			R[i][i]=B;
			if(i+1<n){
				R[i][i+1]=C;
			}
		}
		mat ans=Quick_Pow(R,T)*P;
		for(int i=0;i<n-1;++i){
			printf("%d ",ans[i][0]);
		}
		printf("%d\n",ans[n-1][0]);
	}
	return 0;
}

转载于:https://www.cnblogs.com/autsky-jadek/p/7241654.html