[HDU - 3065 ] 病毒侵袭持续中 [AC自动机- 按输入顺序依次输出文本串中出现过得模板串]

小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下，他发现了网路中的“万恶之源”。这是一个庞大的病毒网站，他有着好多好多的病毒，但是这个网站包含的病毒很奇怪，这些病毒的特征码很短，而且只包含“英文大写字符”。当然小t好想好想为民除害，但是小t从来不打没有准备的战争。知己知彼，百战不殆，小t首先要做的是知道这个病毒网站特征：包含多少不同的病毒，每种病毒出现了多少次。大家能再帮帮他吗？
Input
第一行，一个整数N（1<=N<=1000），表示病毒特征码的个数。
接下来N行，每行表示一个病毒特征码，特征码字符串长度在1—50之间，并且只包含“英文大写字符”。任意两个病毒特征码，不会完全相同。
在这之后一行，表示“万恶之源”网站源码，源码字符串长度在2000000之内。字符串中字符都是ASCII码可见字符（不包括回车）。
Output
按以下格式每行一个，输出每个病毒出现次数。未出现的病毒不需要输出。
病毒特征码: 出现次数
冒号后有一个空格，按病毒特征码的输入顺序进行输出。
Sample Input
3
AA
BB
CC
ooxxCC%dAAAoen…END
Sample Output
AA: 2
CC: 1

分析:　多组数据　卡我一个小时
代码

#include<bits/stdc++.h>
#define N 500010
using namespace std;
queue<int>q;
struct ACT{
	// 字典树相关，分别表示: 存储图，标记单词尾，总节点个数
	int c[N][26], val[N], cnt; 
	int id[N];
	int a[N][26];
	int fail[N]; // 失配转移 
	void init(int n){
		memset(c, 0, sizeof(c[0]) * (n + 3));
		memset(a, 0, sizeof(a[0]) * (n + 3));
		memset(val, 0, sizeof(int) * (n + 3));
		memset(id, 0, sizeof(int) * (n + 3));
		memset(fail, 0, sizeof(int) * (n + 3));
		cnt = 0;
	}
	void ins(char *s, int n){
		int len=strlen(s);int now=0;
		for(int i=0;i<len;i++){
			int v=s[i]-'A';
			if(!c[now][v])c[now][v]=++cnt, a[now][v]=cnt;
			now=c[now][v];
		}
		val[now]++; //标记 是一个单词尾
		id[now] = n;
	}
	void build(){ // fail的建立, 注意fail建立之后　字典树中的指向已经变了
		// 和根节点相连的加入队列中
		for(int i=0;i<26;i++) if(c[0][i])fail[c[0][i]]=0,q.push(c[0][i]);
		while(!q.empty()){
			int u=q.front();q.pop();
			for(int i=0;i<26;i++)
				if(c[u][i]) fail[c[u][i]]=c[fail[u]][i],q.push(c[u][i]);//如果ｕ有ｉ这个分支
				else c[u][i]=c[fail[u]][i]; // 可以跳很多
		}
	}
	void query(char *s){
		int len=strlen(s);
		int now=0;
		for(int i=0;i<len;i++){
			if(s[i] < 'A' || s[i] > 'Z'){
				now = 0; continue;
			}
			now=c[now][s[i]-'A'];
			// val[t] = -1 表示已经访问过, t = 0表示到了根节点
			for(int t=now;t&&~val[t];t=fail[t]) {
				if(val[t]) val[t]++;
			}
		}
	}
}AC;

vector<int>ve;
void dfs(int rt = 0){
	if(AC.val[rt] - 1 > 0) ve.push_back(rt);
	for(int i = 0; i < 26; i++){
		if(AC.a[rt][i]) {
			dfs(AC.a[rt][i]);
		}
	}
}
bool cmp(int a, int b){
	return AC.id[a] < AC.id[b];
}
char mu[1000 + 11][60];
char p[2000000 + 11];
int main(){
	int n; 
	while(~scanf("%d", &n) && n ){
		for(int i = 0; i < n; i++){
			scanf("%s", mu[i]);
			AC.ins(mu[i], i + 1);
		} 
		getchar();
		gets(p);
		AC.build();
		AC.query(p);
		dfs();
		sort(ve.begin(), ve.end(), cmp);
		for(int i = 0; i < (int)ve.size(); i++){
			printf("%s: %d\n", mu[AC.id[ve[i]] - 1], AC.val[ve[i]] - 1);
		} 
		AC.init(AC.cnt + 1);
		ve.clear();
	}
	return 0;
}

代码二

#include<bits/stdc++.h>
#define N 500010
using namespace std;
queue<int>q;
struct ACT{
	// 字典树相关，分别表示: 存储图，标记单词尾，总节点个数
	int c[N][26], val[N], cnt; int id[N];
	int fail[N]; // 失配转移 
	int ans[1000 + 11];
	void init(int n){
		memset(c, 0, sizeof(c[0]) * (n + 3));
		memset(val, 0, sizeof(int) * (n + 3));
		memset(id, 0, sizeof(int) * (n + 3));
		memset(fail, 0, sizeof(int) * (n + 3));
		memset(ans, 0, sizeof(ans));
		cnt = 0;
	}
	void ins(char *s, int n){
		int len=strlen(s);int now=0;
		for(int i=0;i<len;i++){
			int v=s[i]-'A';
			if(!c[now][v])c[now][v]=++cnt;
			now=c[now][v];
		}
		val[now]++; //标记 是一个单词尾
		id[now] = n;
	}
	void build(){ // fail的建立, 注意fail建立之后　字典树中的指向已经变了
		// 和根节点相连的加入队列中
		for(int i=0;i<26;i++) if(c[0][i])fail[c[0][i]]=0,q.push(c[0][i]);
		while(!q.empty()){
			int u=q.front();q.pop();
			for(int i=0;i<26;i++)
				if(c[u][i]) fail[c[u][i]]=c[fail[u]][i],q.push(c[u][i]);//如果ｕ有ｉ这个分支
				else c[u][i]=c[fail[u]][i]; // 可以跳很多
		}
	}
	void query(char *s){
		int len=strlen(s);
		int now=0;
		for(int i=0;i<len;i++){
			if(s[i] < 'A' || s[i] > 'Z'){
				now = 0; continue;
			}
			now=c[now][s[i]-'A'];
			// val[t] = -1 表示已经访问过, t = 0表示到了根节点
			for(int t=now;t&&~val[t];t=fail[t]) {
				if(val[t]) {
					ans[id[t]]++;
				}
			}
		}
	}
}AC;
char mu[1000 + 11][60];
char p[2000000 + 11];
int main(){
	int n; 
	while(~scanf("%d", &n)){
		for(int i = 1; i <= n; i++){
			scanf("%s", mu[i]);
			AC.ins(mu[i], i);
		} 
		getchar();
		gets(p);	
		//scanf("%s", p);
		AC.build();
		AC.query(p);
		for(int i = 1; i <= n; i++){
			if(AC.ans[i]) printf("%s: %d\n", mu[i], AC.ans[i]);
		}
		AC.init(AC.cnt + 1);
	}
	return 0;
}