[FROM WOJ]#3744 JSOI2012 玄武密码

#3744 JSOI2012 玄武密码

题面
在美丽的玄武湖畔，鸡鸣寺边，鸡笼山前，有一块富饶而秀美的土地，人们唤作进香河。相传一日，一缕紫气从天而至，只一瞬间便消失在了进香河中。老人们说，这是玄武神灵将天书藏匿在此。 很多年后，人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是，这份带有玄武密码的文字，与玄武湖南岸台城的结构有微妙的关联。于是，漫长的破译工作开始了。 经过分析，我们可以用东南西北四个方向来描述台城城砖的摆放，不妨用一个长度为N的序列来描述，序列中的元素分别是‘E’，‘S’，‘W’，‘N’，代表了东南西北四向，我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象，分别是东之青龙，西之白虎，南之朱雀，北之玄武，对东南西北四向相对应。 现在，考古工作者遇到了一个难题。对于每一段文字，其前缀在母串上的最大匹配长度是多少呢？

输入
第一行有两个整数，N和M，分别表示母串的长度和文字段的个数。 第二行是一个长度为N的字符串，所有字符都满足是E,S,W和N中的一个。 之后M行，每行有一个字符串，描述了一段带有玄武密码的文字。依然满足，所有字符都满足是E,S,W和N中的一个。

输出
输出有M行，对应M段文字。 每一行输出一个数，表示这一段文字的前缀与母串的最大匹配串长度。

样例输入
7 3
SNNSSNS
NNSS
NNN
WSEE

样例输出
4
2
0

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

SOL
AC自动机模板题
将所有模式串加到一棵Trie树上面，然后getfail。
将母串放到Trie树上去匹配，这里有个优化：如果当前的失配指针之前就被跳过可以结束find，因为你这样会重复走之前的路径。
一个小技巧：这道题只有四个字母，不是26个，转成0,1,2,3方便操作

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+5;
int trans[50];
struct ACtree{
	int ch[N][4],f[N],l[N],pre[N],pos[N],cnt;
	bool flag[N];
	inline int idx(char c){return trans[c-'A'];}
	inline void insert(char *s,int v){
		int len=strlen(s),u=0;l[v]=len;
		for(int i=0;i<len;++i){
			int c=idx(s[i]);
			if(!ch[u][c])ch[u][c]=++cnt,pre[cnt]=u;
			u=ch[u][c];
		}pos[v]=u;
	}
	inline void getfail(){
		queue<int>q;f[0]=0;
		for(int register i=0;i<4;i++)if(ch[0][i])f[ch[0][i]]=0,q.push(ch[0][i]);
		while(!q.empty()){
			int u=q.front();q.pop();
			for(int register i=0;i<4;i++){
				if(!ch[u][i]){ch[u][i]=ch[f[u]][i];continue;}
				q.push(ch[u][i]);f[ch[u][i]]=ch[f[u]][i];
			}
		}
	}
	inline void find(char *t){
		int len=strlen(t),u=0;
		for(int i=0;i<len;++i){
			int c=idx(t[i]);
			u=ch[u][c];	
			int k=u;
			while(k){
				if(flag[k])break;
				flag[k]=1;
				k=f[k];
			}
		}
	}
	inline int work(int v){
		int u=pos[v];
		for(int register i=l[v];i;i--){
			if(flag[u])return i;
			u=pre[u];
		}return 0;
	}
}ac;
int m,n;
char s[N],t[N];
int main(){
	trans['E'-'A']=0,trans['S'-'A']=1,trans['W'-'A']=2,trans['N'-'A']=3;
	scanf("%d%d",&n,&m);
	scanf("%s",s);
	for(int register i=1;i<=m;++i){scanf("%s",t);ac.insert(t,i);}
	ac.getfail();
	ac.find(s);	
	for(int register i=1;i<=m;i++)printf("%d\n",ac.work(i));
	return 0;
}