BZOJ-4327：JSOI2012 玄武密码（AC自动机模板题）

在美丽的玄武湖畔，鸡鸣寺边，鸡笼山前，有一块富饶而秀美的土地，人们唤作进香河。相传一日，一缕紫气从天而至，只一瞬间便消失在了进香河中。老人们说，这是玄武神灵将天书藏匿在此。 

很多年后，人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是，这份带有玄武密码的文字，与玄武湖南岸台城的结构有微妙的关联。于是，漫长的破译工作开始了。 

经过分析，我们可以用东南西北四个方向来描述台城城砖的摆放，不妨用一个长度为N的序列来描述，序列中的元素分别是‘E’，‘S’，‘W’，‘N’，代表了东南西北四向，我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象，分别是东之青龙，西之白虎，南之朱雀，北之玄武，对东南西北四向相对应。 

现在，考古工作者遇到了一个难题。对于每一段文字，其前缀在母串上的最大匹配长度是多少呢？ 

 

 

Input

第一行有两个整数，N和M，分别表示母串的长度和文字段的个数。 

第二行是一个长度为N的字符串，所有字符都满足是E,S,W和N中的一个。 

之后M行，每行有一个字符串，描述了一段带有玄武密码的文字。依然满足，所有字符都满足是E,S,W和N中的一个。 

 

 

Output

输出有M行，对应M段文字。 

每一行输出一个数，表示这一段文字的前缀与母串的最大匹配串长度。 

 

 

Sample Input7 3 SNNSSNS NNSS NNN WSEE

Sample Output4 2 0Hint

 

对于100%的数据，N<=10^7，M<=10^5，每一段文字的长度<=100。

 

应上传者要求，此题不公开，如有异议，请提出.

 

思路：AC自动机，一开始以为不能一直沿着Fail指针上传，会TLE，需要优化。但其实每个点最多标记一次，所以整体复杂度是O(Lc+Ls)的。

所以暴力搞就行了。

#include<bits/stdc++.h>
using namespace std;
const int maxn=10000010;
char c[maxn],s[maxn]; 
int ch[maxn][4],fa[maxn],pos[maxn],vis[maxn];
int q[maxn],dep[maxn],fail[maxn],cnt,head,tail;
int get(char chr){ if(chr=='E') return 0; if(chr=='S') return 1; if(chr=='W') return 2; return 3;}
void insert(int opt)
{
    int Now=0,d=0; 
    for(int i=1;c[i];i++){
        d++;
        if(!ch[Now][get(c[i])]) {
            ch[Now][get(c[i])]=++cnt;
            fa[cnt]=Now; dep[cnt]=d;
        }
        Now=ch[Now][get(c[i])];
    }   pos[opt]=Now;
}
void failbuild()
{
    for(int i=0;i<4;i++) if(ch[0][i]) q[++head]=ch[0][i];
    while(tail<head){
        int Now=q[++tail];
        for(int i=0;i<4;i++){
            if(ch[Now][i]) {
                fail[ch[Now][i]]=ch[fail[Now]][i];
                q[++head]=ch[Now][i];
            }
            else ch[Now][i]=ch[fail[Now]][i];
        }
    }
}
int main()
{
    int L,N,i,j;
    scanf("%d%d%s",&L,&N,s+1);
    for(i=1;i<=N;i++){
        scanf("%s",c+1);
        insert(i);
    }
    failbuild();
    int Now=0;
    for(i=1;i<=L;i++){
        Now=ch[Now][get(s[i])];
        int tmp=Now;
        while(tmp&&!vis[tmp]){
            vis[tmp]=1; tmp=fail[tmp];
        }
    }
    for(i=1;i<=N;i++){
        Now=pos[i];
        while(!vis[Now]&&Now) Now=fa[Now];
        printf("%d\n",dep[Now]);
    }
    return 0;
}