Bzoj 1212: [HNOI2004]L语言(AC自动机+DP)

1212: [HNOI2004]L语言
Time Limit: 10 Sec Memory Limit: 162 MB
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可以被理解的最长前缀的位置。
Sample Input
4 3
is
name
what
your
whatisyourname
whatisyouname
whaisyourname
Sample Output
14
6
0 整段文章’whatisyourname’都能被理解
前缀’whatis’能够被理解
没有任何前缀能够被理解

/*
AC自动机+DP.
令f[i]表示能否匹配到i位置.
如果f[i]能匹配,则f[i-word_len]能匹配.
转移就好. 
*/
#include<queue>
#include<cstdio>
#include<cstring>
#define MAXN 210
using namespace std;
int n,m,len[MAXN],fail[MAXN],f[1100000],tot=1,l,lastmark,pos[MAXN];
struct data{int x[27],b;}tree[MAXN];
bool mark[MAXN];
char s[1100000];
queue<int>q;
void add(int t)
{
    int now=1;l=strlen(s+1);len[t]=l;
    for(int i=1;i<=l;i++)
    {
        int x=s[i]-96;
        if(!tree[now].x[x]) tree[now].x[x]=++tot;
        now=tree[now].x[x];
    }
    tree[now].b=1;pos[now]=t;
    return ;
}
void get_fail()
{
    for(int i=1;i<=26;i++) tree[0].x[i]=1;
    q.push(1);
    while(!q.empty())
    {
        int now=q.front();q.pop(); 
        for(int i=1;i<=26;i++)
        {
            if(!tree[now].x[i]) continue;
            int k=fail[now];
            while(!tree[k].x[i]) k=fail[k];
            k=tree[k].x[i];
            fail[tree[now].x[i]]=k;
            q.push(tree[now].x[i]);
        }
    }
    return ;
}
void Mark()
{
    l=strlen(s);int now=1;
    for(int i=0;i<l;i++)
    {
        int x=s[i]-96;
        while(!tree[now].x[x]) now=fail[now];
        now=tree[now].x[x];
        int k=now;
        while(k)
        {
            f[i+1]|=f[i-len[pos[k]]+1];
            k=fail[k];
        }
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%s",s+1),add(i);
    get_fail();
    while(m--)
    {
        lastmark=0;
        memset(f,0,sizeof f);
        memset(mark,0,sizeof mark);
        scanf("%s",s);
        f[0]=1;Mark();
        for(int i=strlen(s);i>=0;i--)
          if(f[i]) {printf("%d\n",i);break;}
    }
    return 0;
}