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
",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;
}
原文地址:https://www.cnblogs.com/five20/p/9471126.html