【题解】CH3401 石头游戏 矩阵乘法+快速幂

题目链接

描述

石头游戏在一个 n 行 m 列 (1≤n,m≤8) 的网格上进行，每个格子对应一种操作序列，操作序列至多有10种，分别用0~9这10个数字指明。
操作序列是一个长度不超过6且循环执行、每秒执行一个字符的字符串。每秒钟，所有格子同时执行各自操作序列里的下一个字符。序列中的每个字符是以下格式之一：
数字0~9：表示拿0~9个石头到该格子。
NWSE：表示把这个格子内所有的石头推到相邻的格子，N表示上方，W表示左方，S表示下方，E表示右方。
D：表示拿走这个格子的所有石头。
给定每种操作序列对应的字符串，以及网格中每个格子对应的操作序列，求石头游戏进行了 t 秒之后，石头最多的格子里有多少个石头。在游戏开始时，网格是空的。

输入格式

第一行4个整数n, m, t, act。
接下来n行，每行m个字符，表示每个格子对应的操作序列。
最后act行，每行一个字符串，表示从0开始的每个操作序列。

输出格式

一个整数：游戏进行了t秒之后，所有方格中最多的格子有多少个石头。

样例输入

1 6 10 3
011112
1E
E
0

样例输出

3

样例解释

这是另一个类似于传送带的结构。左边的设备0间隔地产生石头并向东传送。设备1向右传送，直到设备2。10秒后，总共产生了5个石头，2个在传送带上，3个在最右边。

---

详见李煜东《算法竞赛进阶指南》

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int n,m,t,act,a[20][20],c[20][20],p;
ll f[70],d[70][70],e[70][70][70],ans;
//状态矩阵f[num(i,j)]记录格子(i,j)中石头的个数 
char s[110],b[20][20];
int num(int i,int j){return (i-1)*m+j;}
void mulself(ll a[70][70],ll b[70][70])
{
    ll c[70][70];
    memset(c,0,sizeof(c));
    for(int i=1;i<=p;i++)
    for(int k=1;k<=p;k++)
    if(a[i][k])for(int j=1;j<=p;j++)c[i][j]+=a[i][k]*b[k][j];
    memcpy(a,c,sizeof(c));
}
void mul(ll a[70],ll b[70][70])
{
    ll c[70];
    memset(c,0,sizeof(c));
    for(int i=1;i<=p;i++)for(int j=1;j<=p;j++)
    c[i]+=a[j]*b[j][i];
    memcpy(f,c,sizeof(c));
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d%d%d",&n,&m,&t,&act);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        for(int j=1;j<=m;j++)a[i][j]=s[j]-'0'+1;
    }
    for(int i=1;i<=act;i++)scanf("%s",&b[i]);
    int x,y;p=n*m+1;
    for(int k=1;k<=60;k++)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            {
                x=a[i][j];y=c[i][j];
                if(isdigit(b[x][y]))
                {
                    e[k][p][num(i,j)]=b[x][y]-'0';
                    e[k][num(i,j)][num(i,j)]=1;
                }
                else if(b[x][y]=='N'&&i>1)e[k][num(i,j)][num(i-1,j)]=1;
                else if(b[x][y]=='W'&&j>1)e[k][num(i,j)][num(i,j-1)]=1;
                else if(b[x][y]=='S'&&i<n)e[k][num(i,j)][num(i+1,j)]=1;
                else if(b[x][y]=='E'&&j<m)e[k][num(i,j)][num(i,j+1)]=1;
                c[i][j]=(y+1)%strlen(b[x]);
            }
            e[k][p][p]=1;
    }
    memcpy(d,e[1],sizeof(e[1]));
    f[p]=1;
    for(int k=2;k<=60;k++)mulself(d,e[k]);
    for(int i=t/60;i;i>>=1)//矩阵快速幂 
    {
        if(i&1)mul(f,d);
        mulself(d,d);
    }
    for(int i=1;i<=t%60;i++)mul(f,e[i]);
    for(int i=1;i<p;i++)ans=max(ans,f[i]);
    printf("%lld\n",ans);
    return 0;
}