【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;
}

转载于:https://www.cnblogs.com/CQzhangyu/p/6265079.html