TZOJ 5986 玄武密码(AC自动机)

匿名 (未验证) 提交于 2019-12-02 23:55:01

描述


输入


之后M行,每行有一个字符串,描述了一段带有玄武密码的文字。依然满足,所有字符都满足是E,S,W和N中的一个。

对于100%的数据,N<=10^7,M<=10^5,每一段文字的长度<=100。

输出


样例输入

7 3
SNNSSNS
NNSS
NNN
WSEE

样例输出

4
2
0

题意

如上。

题解

M个串建AC自动机,由于只有ESWN四个字符,所以只用开4,记录串i在AC自动机最后的位置id,并且记录父节点father和深度d。

查询S所能到达的所有节点,标记vis。

最后串i最后的位置id(叶子节点)往上跑,如果已经被标记输出dis。

时间复杂度O(N+100*M)。

代码

 1 #include<bits/stdc++.h>  2 using namespace std;  3   4 const int N=1e7+5,M=1e5+5;  5 int tot,son[N][4],father[N],fail[N],q[N],d[N],id[M];  6 int n,m;char s[N],s1[N];  7 bool vis[N];  8 inline int get(char ch){  9     if(ch=='E')return 0; 10     else if(ch=='S')return 1; 11     else if(ch=='W')return 2; 12     else return 3; 13 } 14 void insert(int p){ 15     for(int l=strlen(s),x=0,i=0,w;i<l;i++){ 16         if(!son[x][w=get(s[i])])son[x][w]=++tot; 17         father[son[x][w]]=x;d[son[x][w]]=d[x]+1;x=son[x][w]; 18         if(i==l-1)id[p]=x; 19     } 20 } 21 void make(){ 22     int h=1,t=0,i,j,x;fail[0]=-1; 23     for(i=0;i<4;i++)if(son[0][i])q[++t]=son[0][i]; 24     while(h<=t)for(x=q[h++],i=0;i<4;i++) 25         if(son[x][i])fail[son[x][i]]=son[fail[x]][i],q[++t]=son[x][i]; 26         else son[x][i]=son[fail[x]][i]; 27 } 28 void find(){ 29     for(int l=n,x=0,i=0,j;i<l;i++){ 30         x=son[x][get(s1[i])]; 31         for(j=x;j;j=fail[j])vis[j]=1; 32     } 33 } 34 int dfs(int u){ 35     for(int i=u;i>0;i=father[i]) 36         if(d[i]&&vis[i])return d[i]; 37     return 0; 38 } 39 int main() 40 { 41     scanf("%d%d%s",&n,&m,s1); 42     for(int i=1;i<=m;i++)scanf("%s",s),insert(i); 43     make(); 44     find(); 45     for(int i=1;i<=m;i++)printf("%d\n",dfs(id[i])); 46     return 0; 47 }
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!