例题2.23 细胞自动机 LA3740

1.题目描述：点击打开链接

2.解题思路：本题利用矩阵快速幂解决。根据题意可以发现，下一个状态都可以表示成当前状态的线性组合。而这正好可以用矩阵来描述。写出对应的矩阵后我们还会发现一个特点，这个矩阵是一个循环矩阵：把第i行循环向右平移一位即可得到i+1行。这样，我们只需要保存矩阵的第一行即可，计算矩阵乘法的时候也只需要计算第一行即可。时间复杂度为O(n^2logk)，可以承受。

3.代码：

#include<iostream>
#include<algorithm>
#include<cassert>
#include<string>
#include<sstream>
#include<set>
#include<bitset>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<cctype>
#include<complex>
#include<functional>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define rep(i,n) for(int i=0;i<(n);i++)
#define me(s) memset(s,0,sizeof(s))
#define pb push_back
#define lid (id<<1)
#define rid (id<<1|1)

typedef long long ll;
typedef pair<int,int> P;

int MOD;

typedef vector<int> CirculantMatrix;

CirculantMatrix operator*(const CirculantMatrix&a,const CirculantMatrix&b)
{
    int n=a.size();
    assert(b.size()==n);
    CirculantMatrix c(n);
    for(int i=0;i<n;i++) //计算C(0,i)
    {
        c[i]=0;
        for(int j=0;j<n;j++) //累加A(0,j)*B(j,i)
            c[i]=((ll)a[j]*b[(i-j+n)%n]+c[i])%MOD; //第j行是由第0行循环右移j位得到的
    }
    return c;
}


template <typename T>

T fast_pow(const T&a,int k) //计算a^k,要求a支持*运算
{
    assert(k>0);
    if(k==1)return a;
    T b=fast_pow(a,k/2);
    b=b*b;
    if(k&1)b=b*a;
    return b;
}

int main()
{
    int n,m,d,k;
    while(~scanf("%d%d%d%d",&n,&m,&d,&k))
    {
        MOD=m;
        CirculantMatrix mat(n);
        for(int i=0;i<n;i++)
            mat[i]=min(i,n-i)<=d?1:0;
        CirculantMatrix v(n); //把初始状态当做循环矩阵计算
        for(int i=0;i<n;i++)
            scanf("%d",&v[i]);
        v=v*fast_pow(mat,k);
        printf("%d",v[0]);
        for(int i=1;i<n;i++)
            printf(" %d",v[i]);
        puts("");
    }
}