【BZOJ4327】JSOI2012 玄武密码 AC自动机

【BZOJ4327】JSOI2012 玄武密码

Description

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

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

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

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

Input

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

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

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

Output

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

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

Sample Input

7 3
SNNSSNS
NNSS
NNN
WSEE

Sample Output

4
2
0

HINT

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

题解：沿着AC自动机搜几遍即可。

 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
const int maxn=10000010;
struct node
{
    int ch[4],fail,cnt,vis,ans,dep;
}p[maxn];
int dic(char le)
{
    switch(le)
    {
        case 'E':return 0;
        case 'S':return 1;
        case 'W':return 2;
        case 'N':return 3;
    }
}
int n,m,tot;
int Q[maxn],pos[100010];
char w[110],str[maxn];
queue<int> q;
void build()
{
    int i,u,t;
    q.push(1);
    while(!q.empty())
    {
        u=q.front(),q.pop();
        Q[++Q[0]]=u;
        for(i=0;i<4;i++)
        {
            if(p[u].ch[i])
            {
                q.push(p[u].ch[i]);
                t=p[u].fail;
                while(!p[t].ch[i]&&t)    t=p[t].fail;
                if(t)    p[p[u].ch[i]].fail=p[t].ch[i];
                else    p[p[u].ch[i]].fail=1;
            }
        }
    }
}
void search()
{
    int i,j,u=1,t;
    p[u].vis=0;
    for(i=0;i<n;i++)
    {
        while(!p[u].ch[dic(str[i])]&&u)    u=p[u].fail;
        u=p[u].ch[dic(str[i])];
        u=u>0?u:1;
        p[u].vis=1;
    }
    for(i=tot;i>=2;i--)    p[p[Q[i]].fail].vis|=p[Q[i]].vis;
    for(i=1;i<=tot;i++)
    {
        if(p[Q[i]].vis)    p[Q[i]].ans=p[Q[i]].dep;
        for(j=0;j<4;j++)
            if(p[Q[i]].ch[j])
                p[p[Q[i]].ch[j]].ans=p[Q[i]].ans;
    }
    for(i=1;i<=m;i++)    printf("%d\n",p[pos[i]].ans);
}
int main()
{
    scanf("%d%d%s",&n,&m,str);
    int i,j,k,u,t;
    tot=1;
    for(i=1;i<=m;i++)
    {
        scanf("%s",w);
        k=strlen(w);
        u=1;
        for(j=0;j<k;j++)
        {
            if(!p[u].ch[dic(w[j])])    p[u].ch[dic(w[j])]=++tot;
            u=p[u].ch[dic(w[j])];
            p[u].dep=j+1;
        }
        p[u].cnt++;
        pos[i]=u;
    }
    build();
    search();
    return 0;
}