BZOJ 1212: [HNOI2004]L语言 AC自动机

Description

标点符号的出现晚于文字的出现，所以以前的语言都是没有标点的。现在你要处理的就是一段没有标点的文章。 一段文章T是由若干小写字母构成。一个单词W也是由若干小写字母构成。一个字典D是若干个单词的集合。 我们称一段文章T在某个字典D下是可以被理解的，是指如果文章T可以被分成若干部分，且每一个部分都是字典D中的单词。 例如字典D中包括单词{‘is’, ‘name’, ‘what’, ‘your’}，则文章‘whatisyourname’是在字典D下可以被理解的 因为它可以分成4个单词：‘what’, ‘is’, ‘your’, ‘name’，且每个单词都属于字典D，而文章‘whatisyouname’ 在字典D下不能被理解，但可以在字典D’=D+{‘you’}下被理解。这段文章的一个前缀‘whatis’，也可以在字典D下被理解 而且是在字典D下能够被理解的最长的前缀。 给定一个字典D，你的程序需要判断若干段文章在字典D下是否能够被理解。 并给出其在字典D下能够被理解的最长前缀的位置。

Input

输入文件第一行是两个正整数n和m，表示字典D中有n个单词，且有m段文章需要被处理。 之后的n行每行描述一个单词，再之后的m行每行描述一段文章。 其中1<=n, m<=20，每个单词长度不超过10，每段文章长度不超过1M。

Output

对于输入的每一段文章，你需要输出这段文章在字典D可以被理解的最长前缀的位置。

题解：
正常AC自动机建树，然后用一个bool表示这个状态是否可达，AC自动机上每个节点要维护一个指向上一个有节点的fail的指针，我就直接用fail代替了。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<iostream>
#include<iomanip>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
struct Trie
{
    Trie *son[26],*fail;
    int max_len;
    Trie()
    {
        memset(son,0,sizeof(son));
        fail=NULL;
        max_len=0;
    }
}*root=new Trie();
int top=0;
void my_insert(char *s)
{
    Trie *o=root;
    int j;
    for(j=0;s[j];j++)
    {
        if(!o->son[s[j]-'a']) o->son[s[j]-'a']=new Trie();
        o=o->son[s[j]-'a'];
    }
    o->max_len=max(o->max_len,j);
}
void bfs()
{
    static Trie* dui[500000];
    int top=1,my_final=1;
    for(int i=0;i<26;i++)
    {
        if(root->son[i])
        {
            root->son[i]->fail=root;
            dui[my_final++]=root->son[i];
        }
        else root->son[i]=root;
    }
    while(top<my_final)
    {
        Trie *o=dui[top];
        for(int i=0;i<26;i++)
        {
           if(o->son[i])
           {
                o->son[i]->fail=o->fail->son[i];
                dui[my_final++]=o->son[i];
           }
           else o->son[i]=o->fail->son[i];
        }
        top++;
        Trie *t=o->fail;
        while(t!=root && !t->max_len) t=t->fail;
        o->fail=t;
        o->max_len=max(o->max_len,t->max_len);
    }
}
bool f[5000000];
char s[1000000];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        my_insert(s);
    }
    bfs();
    for(int i=1;i<=m;i++)
    {
        memset(f,0,sizeof(f));
        scanf("%s",s+1);
        f[0]=1;
        int ans=0;
        Trie *o=root;
        for(int j=1;s[j];j++)
        {
            o=o->son[s[j]-'a'];
            if(o->max_len)
            {
                Trie *p=o;
                while(p!=root)
                {
                    if(f[j-p->max_len])
                    {
                        f[j]=true;
                        if(j>ans) ans=j;
                        break;
                    }
                    p=p->fail;
                }
            }
        }
        cout<<ans<<endl;
    }

    return 0;
}