Clannad【2018四川省赛】【AC自动机 + DP】

题目链接（第十届四川省赛C题）

---

  挺好的一道题，就是要做一个last优化，每次的last要返回到之前的有值节点，也就是单词的尾的对应节点，然后就不会超时了…… 呜呜呜，之前一直超时，以为是初始化的memset()问题，以前被卡过memset()，然后发现其实要是有多个相同的节点，岂不是会反复无常的跑无用的fail指针，于是就用了last优化掉了，过了。过咯！

  对了，用DP推状态，这个比划一下，挺简单的，就是推对应后缀的去掉后缀的字符串的状态的求和。

---

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1e5 + 7;
const int mod = 1e9 + 7;
int N, M;
char virus[maxN], sss[maxN];
ll dp[maxN];
struct node
{
    node *next[26];
    node *fail, *las;   //las用来寻找上一个为sum==1的点用以优化路径
    int sum, len;
    node()
    {
        for(int i=0; i<26; i++) next[i] = NULL;
        fail = NULL;    las = NULL;
        sum = len = 0;
    }
};
node *root;
void update(char *s)
{
    node *temp = root;
    int len = (int)strlen(s);
    for(int i=0; i<len; i++)
    {
        int x = s[i] - 'a';
        if(temp->next[x] == NULL) temp->next[x] = new node();
        temp = temp->next[x];
    }
    temp->sum = 1;
    temp->len = len;
}
void build_fail()
{
    queue<node *> Q;
    Q.push(root);
    node *temp, *p;
    while(!Q.empty())
    {
        temp = Q.front();   Q.pop();
        for(int i=0; i<26; i++)
        {
            if(temp->next[i])
            {
                if(temp == root)
                {
                    temp->next[i]->fail = root;
                    temp->next[i]->las = root;
                }
                else
                {
                    p = temp->fail;
                    while(p)
                    {
                        if(p->next[i])
                        {
                            temp->next[i]->fail = p->next[i];
                            break;
                        }
                        p = p->fail;
                    }
                    if(!p) temp->next[i]->fail = root;
                    else
                    {
                        if(p->next[i]->sum > 0) temp->next[i]->las = p->next[i];
                        else temp->next[i]->las = p->next[i]->las;
                    }
                }
                Q.push(temp->next[i]);
            }
        }
    }
}
void AC_auto()
{
    node *p = root;
    for(int i=1; i<=N; i++)
    {
        int x = virus[i] - 'a';
        while(p != root && !p->next[x]) p = p->fail;
        p = p->next[x];
        if(!p) p = root;
        node *temp = p;
        while(temp && temp != root)
        {
            if(temp->sum > 0)
            {
                dp[i] += dp[i - temp->len];
                if(dp[i] >= mod) dp[i] -= mod;
            }
            temp = temp->las;
        }
    }
}
inline void init()
{
    for(int i=1; i<=N; i++) dp[i] = 0;
    dp[0] = 1;
    root = (node *)malloc(sizeof(node));
    for(int i=0; i<26; i++) root->next[i] = NULL;
    root->fail = NULL;  root->las = NULL;
    root->sum = 0;
}
int main()
{
    int T;  scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &N, &M);
        init();
        scanf("%s", virus + 1);
        for(int i=1; i<=M; i++)
        {
            scanf("%s", sss);
            update(sss);
        }
        build_fail();
        AC_auto();
        for(int i=1; i<=N; i++) printf("%lld%c", dp[i], i==N?'\n':' ');
    }
    return 0;
}