病毒侵袭持续中 HDU - 3065

#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 =  5e4+10;
int trie[maxn][128]; //字典树
int cntword[maxn];  //记录该单词出现次数
int fail[maxn];     //失败时的回溯指针
int cnt = 0;
int id[maxn];
int ansid[1001]; 
struct node {
	char s[100];
	int id;
} no[1001];
void insertWords(char s[],int len,int index) {
	int root = 0;
	for(int i=0; i<len; i++) {
		int next = s[i];
		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<128; 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<128; 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];
		}
	}
}


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

	while(~scanf("%d",&n)) {
		memset(trie,0,sizeof(trie));
		memset(fail,0,sizeof(fail));
		memset(cntword,0,sizeof(cntword));
		memset(ansid,0,sizeof(ansid));
		memset(no,0,sizeof(no));
		memset(id,0,sizeof(id));
		cnt = 0; 
		for(int i=1; i<=n; i++) {
			scanf("%s",no[i].s) ;//输入单词
			no[i].id = i;
			insertWords(no[i].s,strlen(no[i].s),i);
		}
		fail[0] = 0;
		getFail();
		scanf("%s",s);//查找文本串
		query(s,strlen(s));
		for(int i=1; i<=n; i++) {
			if(ansid[i]) {
				printf("%s: %d\n",no[i].s,ansid[i]);
			}
		}
	}
	return 0;
}
/*
8
.
..
CC
AA
BB
%
C
o
ooxxCC%dAAAAoen....END
3
AA
BB
CC
ooxxCC%dAAAoen....END

*/