LA 3704 Cellular Automaton(矩阵快速幂)

题目 https://vjudge.net/problem/UVALive-3704

构造出来矩阵后,发现这是一个循环矩阵,从第二行开始每一行都是上一行向右,所以只需要保留第一行数据,使时间复杂度边为(O^2log k)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=500+5;
int mod;

struct Matrix{
	int n;
	ll M[maxn];
	Matrix(int n):n(n){memset((M),0,sizeof(M));}
	Matrix operator *(const Matrix & rhs) const{
		Matrix ret(n);
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++)
			ret.M[i]+=M[j]*rhs.M[(j-i+n)%n];
			ret.M[i]%=mod;
		}
			return ret;
	}
	Matrix operator *=(const Matrix &rhs) {
		return *this=*this*rhs;	
	}
};

Matrix pow(Matrix x,int k){
	if(--k==0) return x;
	Matrix ret=x;
	while(k>0){
		if(k&1) ret*=x;
		k>>=1;
		x*=x;
		
	}
	return ret;
}

int main(){
	 int n,d,k;
	while(~scanf("%d %d %d %d",&n,&mod,&d,&k)){
		Matrix A(n),F(n);
		for(int i=0;i<n;i++) scanf("%lld",&F.M[i]);
		for(int i=0;i<n;i++) if(min(i,n-i)<=d) A.M[i]=1;
		A=pow(A,k);
		F=F*A;
		printf("%lld",F.M[0]);
        for(int i=1;i<n;i++) printf(" %lld",F.M[i]);
        printf("\n");
	}
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值