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[N],head,tail,ch[N][5],fail[N],deep[N],isend[100003],f[N],vis[N],mx[100003];  
char s[N],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]);  
}