【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;
}
posted @ 2017-01-09 14:12 CQzhangyu 阅读(...) 评论(...) 编辑 收藏