hdu4758 Walk Through Squares AC自动机（trie图）DP

      提议大致是给n,m，再给两个非空的不相同的只包含R或D字符串s1,s2，问用n个R，m个D可以组成多少种包含s1,s2的字符串。大致思路的话，先根据两个字符串构造trie图，其实就是把失配函数优化掉的ac自动机。之后用dp[i][x][y][k]表示在自动机上状态i，用x个R，y个D，包含s1,s2的状态.令初值dp[0][0][0][0]=1,向上DP，最后统计dp[i][n][m][3]（0<=i<top）的和即可。

     

#include <iostream>#include <algorithm>#include <cstdio>#include <memory.h>#include <string>#include <cstring>#include <queue>using namespace std;typedef long long ll;const int maxn=420;const int tk=2;int tree[maxn][tk];int val[maxn];int last[maxn];int f[maxn];int n,m,p,q,k;int top;int dp[220][110][110][4];int ok[maxn];const int mod=1000000007;inline int idx(char s){    if (s=='R') return 0;    return 1;}struct autoAC{    void init()    {        top=1;        tree[0][0]=tree[0][1]=0;        memset(val,0,sizeof val);        memset(f,0,sizeof f);        memset(last,0,sizeof last);    }    void insert(char* s,int rank)    {        int rt,nxt;        for (rt=0; *s; rt=nxt,++s)        {            nxt=tree[rt][idx(*s)];            if (nxt==0)            {                tree[rt][idx(*s)]=nxt=top;                tree[top][0]=tree[top][1]=0;                top++;            }        }        val[rt]=rank;    }    void getFail()    {        queue<int> q;        f[0]=0;        for (int c=0; c<tk; c++)        {            int u=tree[0][c];            if (u)            {                q.push(u);                f[u]=0;                last[u]=0;            }        }        while(!q.empty())        {            int r=q.front();            q.pop();            for (int c=0; c<tk; c++)            {                int u=tree[r][c];                if (!u)                {                    tree[r][c]=tree[f[r]][c];                    continue;                }                q.push(u);                int v=f[r];                while(v && !tree[v][c]) v=f[v];                f[u]=tree[v][c];                last[u]=val[f[u]]? f[u] : last[f[u]];            }        }    }}ac;int tt;int mx,my;char s[105],ss[105],str[105];inline void work(){    memset(ok,0,sizeof ok);    for (int i=0; i<top; i++)    {        int j=last[i];        ok[i]=val[i];        while(j)        {            ok[i]=ok[i]|val[j];            j=last[j];        }    }    dp[0][0][0][0]=1;    int sta;    for (int x=0; x<=mx; x++)     for (int y=0; y<=my; y++)     for (int i=0; i<top; i++)     for (int k=0; k<4; k++)     {         if (dp[i][x][y][k]==0) continue;         if (x<mx)         {             int nxt=tree[i][0];              sta=k|ok[nxt];             dp[nxt][x+1][y][sta]+=dp[i][x][y][k];             if (dp[nxt][x+1][y][sta]>=mod) dp[nxt][x+1][y][sta]-=mod;         }         if (y<my)         {             int nxt=tree[i][1];              sta=k|ok[nxt];             dp[nxt][x][y+1][sta]+=dp[i][x][y][k];             if (dp[nxt][x][y+1][sta]>=mod) dp[nxt][x][y+1][sta]-=mod;         }     }     ll res=0;     for (int i=0; i<top; i++)     {         res+=dp[i][mx][my][3];//         if (res>=mod) res-=mod;     }     res%=mod;     cout<<res<<endl;}int main(){//    freopen("in.txt","r",stdin);    scanf("%d",&tt);    while(tt--)    {        ac.init();        scanf("%d%d",&mx,&my);        for (int i=1; i<=2; i++)        {            scanf("%s",s);            ac.insert(s,i);        }        ac.getFail();        memset(dp,0,sizeof dp);        work();    }    return 0;}