TZOJ 5986 玄武密码(AC自动机)
描述 输入 之后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