HDU - 3065 病毒侵袭持续中 AC自动机

题目链接：https://vjudge.net/problem/HDU-3065

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>
using namespace std;
const int N = 1e5 + 10;
int n, m;
int vis[N];
vector<int > ans;
char st[1010][55];
struct AC {
	int tree[N][130];
	int fail[N];
	int val[N];
	int tot;
	void init(int x) {
		tot = 0;
		for(int i = 0; i <= x; i++) {
			fail[i] = 0;
			val[i] = 0;
			for(int j = 0; j < 130; j++)
				tree[i][j] = 0;
		}
	}
	void insert(int x, char *str) {
		int len = strlen(str);
		int p = 0, op;
		for(int i = 0; i < len; i++) {
			op = str[i] ;
			if(tree[p][op] == 0)
				tree[p][op] = ++tot;
			p = tree[p][op];
		}
		val[p] = x;
	}
	void getfail() {
		queue<int> q;
		for(int i = 0; i < 130; i++) {
			if(tree[0][i]) {
				fail[tree[0][i]] = 0;
				q.push(tree[0][i]);
			}
		}
		int now;
		while(!q.empty()) {
			now = q.front(); q.pop();
			for(int i = 0; i < 130; i++) {
				if(tree[now][i]) { // 有这个节点，失配后指向父亲失配后指向的那个节点的下一个 
					fail[tree[now][i]] = tree[fail[now]][i];
					q.push(tree[now][i]);
				} else { // 没有这个节点，直接指向失配后指向的那个节点 
					tree[now][i] = tree[fail[now]][i];
				}
			}
		}
	}
	int query(char *s) {
	    memset(vis, 0, sizeof(vis));
	    ans.clear();
	    int flag = 0;
	    int p = 0, res = 0;
	    int len = strlen(s);
	    for(int i = 0; i < len; i++) {
	        p = tree[p][s[i]];
	        for(int j = p; j ; j = fail[j]) { // 不断找fail 
	            if(val[j]) {
	            	vis[val[j]]++;
				}
	        }
	    }
	    for(int i = 1; i <= n; i++) 
	    	if(vis[i])
	    		printf("%s: %d\n", st[i], vis[i]);
	}
}A;
int main() {
	char s[N * 20];
	while(~scanf("%d", &n)) {
		A.init(n * 50);
		for(int i = 1; i <= n; i++) {
			scanf("%s", st[i]);
			A.insert(i, st[i]);
		}	
		A.getfail();
		scanf("%s", s);
		A.query(s);
	}
	
	
	return 0;
}