UVALive 3704 细胞自动机 矩阵快速幂

是时候要做做数学类的题目了

这属于比较简单的矩阵快速幂了，因为有个已知的矩阵循环的结论，所以为了节约时空，只需要保留一行即可，这个稍微有点难写，也不是难写，主要是注意细节。其他的矩阵快速幂一下即可

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 505
#define ll long long
using namespace std;
struct Mat
{
    ll mat[][N];
}E,Z,O;
ll n,m,d,k;
void init()
{
    memset(E.mat,,sizeof E.mat);
    for (int i=;i<n;i++)
        E.mat[i][i]=;
    memset(O.mat,,sizeof O.mat);
    int tmp=;
    for (int j=-d;tmp<*d+;j++,tmp++){
        if (j<) j+=n;
        if (j>=n) j%=n;
        O.mat[][j]=;
    }
//    cout<<"!!!"<<endl;
//    for (int i=0;i<n;i++)
//        cout<<O.mat[0][i]<<" ";
//    cout<<endl;
}
Mat operator *(Mat a,Mat b)
{
    Mat c=Z;
    int i,j;
    for (i=;i<n;i++){
        int cur=n-i;
        for (j=;j<n;j++){
          if (cur>=n) cur%=n;//因为是循环矩阵，由第一行即可推得其他列的情况
          c.mat[][i]+=a.mat[][j]*b.mat[][cur];
          cur++;
          if (c.mat[][i]>=m) c.mat[][i]%=m;
        }
    }
    return c;
}
Mat operator ^(Mat a,ll x)
{
    Mat c=E;
    for (;x;x>>=){
        if (x&){
            c=c*a;
        }
        a=a*a;
    }
    return c;
}
ll A[N],B[N];
int main()
{
    memset(Z.mat,,sizeof Z.mat);
    while (scanf("%lld%lld%lld%lld",&n,&m,&d,&k)!=EOF)
    {
        init();
        Mat ans=O^k;
//        for (int i=0;i<n;i++)
//            cout<<ans.mat[0][i]<<" ~ ";
//        cout<<endl;
        for (int i=;i<n;i++)
            scanf("%lld",&A[i]);
        for (int i=;i<n;i++){
            int cur=n-i;
            B[i]=;
            for (int j=;j<n;j++){
                if (cur>=n) cur%=n;
                B[i]+=ans.mat[][cur]*A[j];
                cur++;
                if (B[i]>=m) B[i]%=m;
            }
        }
        printf("%lld",B[]);
        for (int i=;i<n;i++)
            printf(" %lld",B[i]);
        printf("\n");
    }
    return ;
}