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

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自动机.

         将母串放到AC自动机上匹配,

         匹配到一个点就沿着它的fail指针向上走,将经过的点标记成已匹配,碰到已匹配的点就退出.

         这样每个点最多只会被标记一次.

         统计答案的时候从每个单词的最后一个点向上走到第一个被标记的节点即可.

代码;

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 10000010
#define M 110
using namespace std;
int a[N][4],l[N],cnt=1,n,m,q[N],fail[N],fa[N],pos[N],point[N];
char ch[M],s[N];
int cal(char x){
  if (x=='E') return 0;
  if (x=='S') return 1;
  if (x=='W') return 2;
  if (x=='N') return 3;
}
void insert(int x){
  l[x]=strlen(ch);int now=1;
  for (int i=0;i<l[x];i++){
    int u=cal(ch[i]);
    if (a[now][u]) now=a[now][u];
    else a[now][u]=++cnt,fa[a[now][u]]=now,now=a[now][u];
  } 
  point[x]=now;
}
void getfail(){
  int h(0),t(1);
  q[0]=1;fail[1]=0;
  while (h<t){
    int u=q[h++];
    for (int i=0;i<4;i++){
      int k=fail[u];
      while (!a[k][i]) k=fail[k];
      if (a[u][i]){
        fail[a[u][i]]=a[k][i];
        q[t++]=a[u][i];
      }
      else a[u][i]=a[k][i];
    }
  }
}
void pre(){
  int now=1;
  for (int i=0;i<n;i++){
    int u=cal(s[i]);
    now=a[now][u];
    for (int j=now;j;j=fail[j])
      if (pos[j]) break;
      else pos[j]=1;
  }
} 
int getans(int x){
  int ans=l[x];
  for (int i=point[x];i;i=fa[i]){
    if (pos[i]) return ans;
    ans--;
  } 
  return 0;
} 
int main(){ 
  //freopen("a.in","r",stdin);
  scanf("%d%d",&n,&m);
  scanf("%s",s);
  for (int i=1;i<=m;i++){
    scanf("%s",ch);
    insert(i);
  } 
  for (int i=0;i<4;i++) a[0][i]=1;
  getfail();
  pre();
  for (int i=1;i<=m;i++) printf("%d\n",getans(i)); 
}