HDU 6086 Rikka with String ——（AC自动机 + DP）

　　这是一个AC自动机+dp的问题，在中间的串的处理可以枚举中断点来插入自动机内来实现，具体参见代码。

　　在这题上不止为何一直MLE，一直找不到结果（lyf相同写法的代码消耗内存较少），还好考虑到这题节点应该不会过多，可以少开一点节点数。

　　代码如下：

#include <bits/stdc++.h>
using namespace std;
//const int N = 5e4 + 5;
//const int MAX_Tot = 6 * 20 * 20 + 6 * 20 + 100;
const int MAX_Tot = 500;
typedef long long ll;
const int mod = 998244353;

int T, n, L;
void add(int &a, int b)
{
    a += b;
    if(a >= mod) a -= mod;
    if(a < 0) a += mod;
}
struct Aho
{
    struct state
    {
        int nxt[2];
        int fail;
        int ed, vis;
    }stateTable[MAX_Tot];

    int size;

    queue<int> que;

    void init()
    {
        while(que.size()) que.pop();
        for(int i=0;i<MAX_Tot;i++)
        {
            memset(stateTable[i].nxt,0,sizeof(stateTable[i].nxt));
            stateTable[i].fail = stateTable[i].ed = stateTable[i].vis = 0;
        }
        size = 1;
    }

    void insert(char *s,int which,int type)
    {
        int n = strlen(s);
        int now = 0;
        for(int i=0;i<n;i++)
        {
            char c = s[i];
            if(!stateTable[now].nxt[c-'0'])
                stateTable[now].nxt[c-'0'] = size++;
            now = stateTable[now].nxt[c-'0'];
        }
        if(type == 1) stateTable[now].ed |= (1<<which);
        else stateTable[now].vis |= (1<<which);
    }

    void build()
    {
        stateTable[0].fail = -1;
        que.push(0);

        while(que.size())
        {
            int u = que.front();que.pop();
            for(int i=0;i<2;i++)
            {
                if(stateTable[u].nxt[i])
                {
                    if(u == 0) stateTable[stateTable[u].nxt[i]].fail = 0;
                    else
                    {
                        int v = stateTable[u].fail;
                        while(v != -1)
                        {
                            if(stateTable[v].nxt[i])
                            {
                                stateTable[stateTable[u].nxt[i]].fail = stateTable[v].nxt[i];
                                stateTable[stateTable[u].nxt[i]].ed |= stateTable[stateTable[stateTable[u].nxt[i]].fail].ed;
                                stateTable[stateTable[u].nxt[i]].vis |= stateTable[stateTable[stateTable[u].nxt[i]].fail].vis;
                                break;
                            }
                            v = stateTable[v].fail;
                        }
                        if(v == -1) stateTable[stateTable[u].nxt[i]].fail = 0;
                    }
                    que.push(stateTable[u].nxt[i]);
                }
                /*****建立自动机nxt指针*****/
                else
                {
                    if(u == 0) stateTable[u].nxt[i] = 0;
                    else
                    {
                        int p = stateTable[u].fail;
                        while(p != -1 && stateTable[p].nxt[i] == 0) p = stateTable[p].fail;
                        if(p == -1) stateTable[u].nxt[i] = 0;
                        else stateTable[u].nxt[i] = stateTable[p].nxt[i];
                    }
                }
                /*****建立自动机nxt指针*****/
            }
        }
    }

    int dp[110][MAX_Tot][1<<6];
    void solve()
    {
        memset(dp,0,sizeof dp);
        dp[0][0][0] = 1;
        for(int i=1;i<=L;i++)
        {
            for(int j=0;j<size;j++)
            {
                for(int k=0;k<(1<<n);k++)
                {
                    for(int ch=0;ch<2;ch++)
                    {
                        int now = stateTable[j].nxt[ch];
                        int nowS = stateTable[now].ed;
                        add(dp[i][now][k|nowS], dp[i-1][j][k]);
                    }
                }
            }
        }
        int ans = 0;
        for(int i=0;i<size;i++)
        {
            for(int j=0;j<(1<<n);j++)
            {
                if((stateTable[i].vis | j) == (1<<n)-1)
                {
                    add(ans, dp[L][i][j]);
                }
            }
        }
        printf("%d
",ans);
    }
}aho;
char s[50],t[50];
void work(int fenge,int which)
{
    int f = fenge;
    int you = fenge + 1;
    bool can = 1;
    int len = strlen(s);
    while(1)
    {
        if(s[fenge] == s[you])
        {
            can = 0;
            break;
        }
        fenge--; you++;
        if(fenge < 0 || you >= len) break;
    }
    if(!can) return ;
    fenge = f;
    if((fenge+1) * 2 <= len)
    {
        for(int i=0;i<len-(fenge+1);i++) t[i] = s[len-1-i] == '1' ? '0' : '1'; t[len-(fenge+1)] = 0;
        aho.insert(t,which,2);
    }
    else
    {
        for(int i=0;i<=fenge;i++) t[i] = s[i]; t[fenge+1] = 0;
        aho.insert(t,which,2);
    }
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        aho.init();
        scanf("%d%d",&n,&L);
        for(int i=0;i<n;i++)
        {
            scanf("%s",s);
            int len = strlen(s);
            aho.insert(s,i,1);
            for(int j=0;j<len;j++) t[j] = s[len-j-1] == '1' ? '0' : '1'; t[len] = 0;
            aho.insert(t,i,1);
            for(int j=0;j<len-1;j++) work(j,i);
        }
        aho.build();
        aho.solve();
    }
    return 0;
}