Day2 - C - One-Dimensional Cellular Automaton

题目分析

如果将题目中原先的序列，我们可以通过一个构造转换矩阵来实现对原先序列的一个转换，转换如下，如果n为5的话：

S(0,T-1)       B C 0 0 0       S(0,T)

S(1,T-1)       A B C 0 0          S(1,T)

S(2,T-1)   X   0 AB C 0  =  S(2,T)

S(3,T-1)       0 0 A B C       S(3,T)

S(4,T-1)       0 0 0 A B       S(4,T)

这样转化后就变成一个求某矩阵的T次幂，然后再做一次矩阵乘法就可以了，T很大，但是可以通过快速幂来实现。

做题时，出现的失误是没有考虑格式，即每行末尾没空格。

AC代码

#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <iostream>
using namespace std;

int n,m,a,b,c,t;

struct matrix
{
       int a[50][50];
}res,origin;

int num[50];


matrix multiply(matrix x,matrix y)
{
       matrix temp;
       memset(temp.a,0,sizeof(temp.a));
       int i,j,k;
       for(i=0;i<n;i++)
       {
           for(j=0;j<n;j++)
           {
               for(k=0;k<n;k++)
               {
                    temp.a[i][j]+=x.a[i][k]*y.a[k][j];
                    temp.a[i][j]%=m;
               }
           }
       }
       return temp;
}

void calc(int T)
{
    while(T)
    {
         if(T&1)
            res=multiply(res,origin);
         T>>=1;
            origin=multiply(origin,origin);
    }
}

void init()
{
     memset(origin.a,0,sizeof(origin.a));
     memset(res.a,0,sizeof(res.a));
     cin>>m>>a>>b>>c>>t;
     int i,j;
     origin.a[0][0]=b;
     origin.a[0][1]=c;
     origin.a[n-1][n-2]=a;
     origin.a[n-1][n-1]=b;
     res.a[0][0]=res.a[n-1][n-1]=1;
     for (i=1;i<n-1;i++)
     {
         res.a[i][i]=1;
         origin.a[i][i-1]=a;
         origin.a[i][i]=b;
         origin.a[i][i+1]=c;
     }
     calc(t);
     for (i=0;i<n;i++)
         scanf("%d",&num[i]);
     for (i=0;i<n;i++)
     {
         int ans=0;
         for (j=0;j<n;j++)
         {
             ans+=num[j]*res.a[i][j];
             ans%=m;
         }
         if(i!=0)
            printf(" ");
         printf("%d",ans);
     }
     cout<<endl;
}

int main()
{
    while((cin>>n) && n!=0)
    {
        init();
    }
    return 0;
}