HDU 4758 Walk Through Squares （2013南京网络赛1011题，AC自动机+DP）

题意：

有一个n×m的矩阵，有两种走法，每种走法都是一个包含D和R的序列，D表示向下走，R表示向右走，问从（1，1）走到（n，m）有多少方案是包含这两种走法的

题解：

我们用这两种走法建立AC自动机，然后在AC自动机上DP解决方案数问题，设$dp[i][j][k][S]$表示当前走了i个D，j个R，对应在AC自动机上的k点，此时已经包含的走法集合为S

直接四个for循环枚举每一维统计方案数

最后只统计包含两种走法的方案即可

---

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 105;
const int MOD = 1000000007;
char s[MAXN];
int nxt[MAXN<<1][2],fail[MAXN<<1],sta[MAXN<<1],tot;
int dp[MAXN][MAXN][MAXN<<1][4];
inline void Init(){
    for(int i=0;i<=tot;i++){
        fail[i]=sta[i]=0;
        for(int j=0;j<2;j++) nxt[i][j]=0;
    }
    tot=0;
}
inline int idx(char ch){ if(ch=='R') return 0; else return 1; }
inline void Insert(char *s,int k){
    int rt=0,len=strlen(s);
    for(int i=0;i<len;i++){
        if(!nxt[rt][idx(s[i])]) nxt[rt][idx(s[i])]=++tot;
        rt=nxt[rt][idx(s[i])];
    }
    sta[rt]|=(1<<k);
}
inline void Build(){
    queue<int> que;
    for(int i=0;i<2;i++) if(nxt[0][i]) que.push(nxt[0][i]);
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=0;i<2;i++){
            if(nxt[u][i]){
                fail[nxt[u][i]] = nxt[fail[u]][i];
                que.push(nxt[u][i]);
                sta[nxt[u][i]] |= sta[nxt[fail[u]][i]];
            }else{
                nxt[u][i] = nxt[fail[u]][i];
            }
        }
    }
}
inline void add(int &x,int y){ x = (x+y)%MOD; }
int main(){
    int T; scanf("%d",&T);
    while(T--){
        Init();
        int n,m; scanf("%d%d",&n,&m);
        for(int i=0;i<2;i++) scanf("%s",s),Insert(s,i);
        Build();
        memset(dp,0,sizeof(dp));
        dp[0][0][0][0]=1;
        for(int i=0;i<=n;i++){
            for(int j=0;j<=m;j++){
                for(int k=0;k<=tot;k++){
                    for(int p=0;p<4;p++){
                        if(!dp[i][j][k][p]) continue;
                        add(dp[i+1][j][nxt[k][0]][p|sta[nxt[k][0]]],dp[i][j][k][p]);
                        add(dp[i][j+1][nxt[k][1]][p|sta[nxt[k][1]]],dp[i][j][k][p]);
                    }
                }
            }
        }
        int res=0;
        for(int i=0;i<=tot;i++) add(res,dp[n][m][i][3]);
        printf("%d\n",res);
    }
    return 0;
}