bzoj 4327: JSOI2012 玄武密码 （AC自动机）

4327: JSOI2012 玄武密码

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 204  Solved: 85

[Submit][Status][Discuss]

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。

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

Source

[Submit][Status][Discuss]

题解：AC自动机

对于询问串建立ＡＣ自动机，然后用模板串在上面进行匹配。

最后查询每个串从结束点到根路径上有访问标记的最深的点的深度。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 10000007
using namespace std;
int n,m,cnt,q
,head,tail,ch
[5],fail
,deep
,isend[100003],f
,vis
,mx[100003];
char s
,ch1[203];
int get(char c)
{
if (c=='N') return 1;
if (c=='S') return 2;
if (c=='W') return 3;
if (c=='E') return 4;
}
void build(int x)
{
int len=strlen(ch1+1); int now=0;
for (int i=1;i<=len;i++){
int x=get(ch1[i]);
if (!ch[now][x]) ch[now][x]=++cnt;
deep[ch[now][x]]=deep[now]+1;
f[ch[now][x]]=now;
now=ch[now][x];
}
isend[x]=now;
}
void make_fail()
{
int now=0; head=0,tail=0;
for (int i=1;i<=4;i++)
if (ch[now][i]) q[++tail]=ch[now][i];
while (head<tail) {
int now=q[++head];
for (int i=1;i<=4;i++) {
if (!ch[now][i]) {
ch[now][i]=ch[fail[now]][i];
continue;
}
fail[ch[now][i]]=ch[fail[now]][i];
q[++tail]=ch[now][i];
}
}
//for (int i=1;i<=cnt;i++) cout<<fail[i]<<" ";
//cout<<endl;
}
void solve()
{
int now=0;
for (int i=1;i<=n;i++) {
int x=get(s[i]);
now=ch[now][x];  vis[now]=1;
int k=now;
while (fail[k]&&!vis[fail[k]])
vis[fail[k]]=1,k=fail[k];
}
//for (int i=1;i<=cnt;i++) cout<<vis[i]<<" ";
//cout<<endl;
for (int i=1;i<=m;i++){
for (int j=isend[i];j;j=f[j])
if (vis[j]) {
mx[i]=deep[j];
break;
}
}
}
int main()
{
freopen("a.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%d%d",&n,&m);
scanf("%s",s+1);
for (int i=1;i<=m;i++) {
scanf("%s",ch1+1);
build(i);
}
make_fail();
solve();
for (int i=1;i<=m;i++) printf("%d\n",mx[i]);
}