AI智能
改变未来

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#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[4],cnt,ed,fail,dep,pre;char t[N*10],s;bool vis;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

[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;}

[p] 

赞(0) 打赏
未经允许不得转载:爱站程序员基地 » BZOJ 4327: JSOI2012 玄武密码