BZOJ 4327: 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。

 

Solution：

　　本题AC自动机+简单搜索就好了。

　　存下主串，再对所有模式串构建AC自动机（记录下每个串结尾的节点和每个节点的前趋以及深度），然后直接在trie树中查询主串，对于扫到的节点p，标记所有没访问过的fail[p]节点（访问过的就不用再往上标记了），然后直接从每个模式串的尾节点往上dfs，搜到的第一个被标记的节点的深度就是答案了。

代码：

#include<bits/stdc++.h>
#define il inline
#define ll long long
#define For(i,a,b) for(int (i)=(a);(i)<=(b);(i)++)
#define Bor(i,a,b) for(int (i)=(b);(i)>=(a);(i)--)
using namespace std;
const int N=1000005;
int n,m,trie[N][4],cnt,ed[N],fail[N],dep[N],pre[N];
char t[N*10],s[N];
bool vis[N];

il int id(char c){
    if(c=='E')return 0;
    if(c=='S')return 1;
    if(c=='W')return 2;
    return 3;
}

il void insert(char *s,int num){
    int len=strlen(s),p=0,x;
    For(i,0,len-1){
        x=id(s[i]);
        if(!trie[p][x])dep[++cnt]=dep[p]+1,trie[p][x]=cnt,pre[cnt]=p;
        p=trie[p][x];
    }
    ed[num]=p;
}

il void bfs(){
    queue<int>q;
    For(i,0,3)if(trie[0][i])q.push(trie[0][i]);
    while(!q.empty()){
        int u=q.front();q.pop();
        For(i,0,3) {
            int v=trie[u][i];
            if(v) fail[v]=trie[fail[u]][i],q.push(v);
            else trie[u][i]=trie[fail[u]][i];
        }
    }
}

il int dfs(int now){
    if(vis[now])return dep[now];
    if(!now) return 0;
    dfs(pre[now]);
}

il void search(char *s){
    int len=strlen(s),p=0;
    For(i,0,len-1){
        p=trie[p][id(s[i])];
        for(int j=p;j&&!vis[j];j=fail[j]) vis[j]=1;
    }
    For(i,1,m) printf("%d\n",dfs(ed[i]));
}

int main(){
    scanf("%d%d%s",&n,&m,t);
    For(i,1,m) scanf("%s",s),insert(s,i);
    bfs();
    search(t);
    return 0;
}