病毒侵袭 HDU - 2896 AC 自动机

#include <queue>
#include <cstdlib>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn =  1e5+10;
int trie[maxn][95]; //字典树
//int cntword[maxn];  //记录该单词出现次数
bool vis[maxn];
int fail[maxn];     //失败时的回溯指针
int cnt = 0;
int id[maxn];
int ansid[5],num;
void insertWords(char s[],int len,int index) {
	int root = 0;
	for(int i=0; i<len; i++) {
		int next = s[i]-33;
		if(!trie[root][next])
			trie[root][next] = ++cnt;
		root = trie[root][next];
	}
//	cntword[root]++;      //当前节点单词数+1
	id[root] = index;
}
void getFail() {
	queue <int>q;
	for(int i=0; i<95; i++) {   //将第二层所有出现了的字母扔进队列
		if(trie[0][i]) {
			fail[trie[0][i]] = 0;
			q.push(trie[0][i]);
		}
	}

//fail[now]    ->当前节点now的失败指针指向的地方
////tire[now][i] -> 下一个字母为i+'a'的节点的下标为tire[now][i]
	while(!q.empty()) {
		int now = q.front();
		q.pop();

		for(int i=0; i<95; i++) {   //查询26个字母
			if(trie[now][i]) {
				//如果有这个子节点为字母i+'a',则
//让这个节点的失败指针指向(((他父亲节点)的失败指针所指向的那个节点)的下一个节点)
				//有点绕,为了方便理解特意加了括号

				fail[trie[now][i]] = trie[fail[now]][i];
				q.push(trie[now][i]);
			} else //否则就让当前节点的这个子节点
				//指向当前节点fail指针的这个子节点
				trie[now][i] = trie[fail[now]][i];
		}
	}
}


int query(char s[],int len) {
	int now = 0,ans = 0;
	for(int i=0; i<len; i++) { //遍历文本串
		now = trie[now][s[i]-33];  //从s[i]点开始寻找
		for(int j=now; j && vis[j]!=1; j=fail[j]) {
			//一直向下寻找,直到匹配失败(失败指针指向根或者当前节点已找过).
			//ans += cntword[j];
			if(id[j]!=0)ans++,ansid[num++] = id[j],vis[j]=1;   //将遍历过后的结点标记,防止重复计算
		}
	}
	return ans;
}
char s[maxn];
int main() {
	int n,m,nt=1;
	cnt = 0;

	scanf("%d",&n);
	for(int i=0; i<n; i++) {
		scanf("%s",s) ;//输入单词
		insertWords(s,strlen(s),i+1);
	}
	fail[0] = 0;
	getFail();
	scanf("%d",&m);
	int nn = 0;
	for(int i=0; i<m; i++) {
		memset(vis,0,sizeof(vis));
		scanf("%s",s);//查找文本串
		num = 0;
		num = query(s,strlen(s));
		if(num>0) {
			printf("web %d:",i+1);
			sort(ansid,ansid+num);
			for(int i=0; i<num; i++) {
				printf(" %d",ansid[i]);
			}
			nn++;
			printf("\n");
		}

	}
	printf("total: %d\n",nn);

	return 0;
}
/*
3
bbb
aaa
ccc
5
aa
aaabbbccc
bbb
cccaaabbb
aaabbbccc

*/