HDU-2896 病毒侵襲 （AC自動機，模板題變形）

題目鏈接：HDU - 2896

病毒侵襲

當太陽的光輝逐漸被月亮遮蔽，世界失去了光明，大地迎來最黑暗的時刻。。。。在這樣的時刻，人們卻異常興奮——我們能在有生之年看到500年一遇的世界奇觀，那是多麼幸福的事兒啊
但網路上總有那麼些網站，開始藉着民衆的好奇心，打着介紹日食的旗號，大肆傳播病毒。小t不幸成爲受害者之一。小t如此生氣，他決定要把世界上所有帶病毒的網站都找出來。當然，誰都知道這是不可能的。小t卻執意要完成這不能的任務，他說：“子子孫孫無窮匱也！”（愚公後繼有人了）。
萬事開頭難，小t收集了好多病毒的特徵碼，又收集了一批詭異網站的源碼，他想知道這些網站中哪些是有病毒的，又是帶了怎樣的病毒呢？順便還想知道他到底收集了多少帶病毒的網站。這時候他卻不知道何從下手了。所以想請大家幫幫忙。小t又是個急性子哦，所以解決問題越快越好哦

Input

第一行，一個整數N（1<=N<=500），表示病毒特徵碼的個數。
接下來N行，每行表示一個病毒特徵碼，特徵碼字符串長度在20—200之間。
每個病毒都有一個編號，依此爲1—N。
不同編號的病毒特徵碼不會相同。
在這之後一行，有一個整數M（1<=M<=1000），表示網站數。
接下來M行，每行表示一個網站源碼，源碼字符串長度在7000—10000之間。
每個網站都有一個編號，依此爲1—M。
以上字符串中字符都是ASCII碼可見字符（不包括回車）。

Output

依次按如下格式輸出按網站編號從小到大輸出，帶病毒的網站編號和包含病毒編號，每行一個含毒網站信息。
web 網站編號: 病毒編號 病毒編號 …
冒號後有一個空格，病毒編號按從小到大排列，兩個病毒編號之間用一個空格隔開，如果一個網站包含病毒，病毒數不會超過3個。
最後一行輸出統計信息，如下格式
total: 帶病毒網站數
冒號後有一個空格。

Sample Input

3
aaa
bbb
ccc
2
aaabbbccc
bbaacc

Sample Output

web 1: 1 2 3
total: 1

題意

n個模式串，是病毒的特徵碼，特徵碼各不相同。
m個主串，是網站的內容，問你這個網站包括多少病毒，也就是內容能匹配上幾個模式串，匹配了哪幾個模式串。

我的理解

串的長度不大，但是也不小，肯定用暴力做kmp是不可以的，我們上AC自動機。
對於每個主串，我們用AC自動機做一次匹配，我們將match函數稍微一改，返回值不是一個串的數量，而是存到一個vector中去，完成一次匹配之後，我們將vector中的病毒編號排序，保存一下，最後打印出來。

這個題我一開始認爲是一個板子題的改版，就大意了。有好幾個很大的坑。
1.題目中說以上字符串中字符都是ASCII碼可見字符（不包括回車），如果默認只有字母，對不起，RE。
2.web的輸出末尾不得有空格，total的行末加回車，沒注意到，對不起，PE。
3.要求web後的病毒編號由大到小排序，如果你沒注意到，對不起，WA。

代碼

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10010;
const int maxt = 58010;

vector<int> tmp;

struct  Aho_Corasick{
	struct Trie
	{
		int next[130];
		int fail,cnt;
	}trie[maxt];

	int size;
	queue<int> q;

	void init(){
		while(q.size())	q.pop();
		size = 1;
		for(int i = 0;i<maxt;i++){
			memset(trie[i].next,0,sizeof(trie[i].next));
			trie[i].fail = trie[i].cnt = 0;
		}
	}

	void insert(char *s,int no){
		int len = strlen(s);
		int now = 0;
		for(int i = 0;i<len;i++){
			int id = s[i];
			if(trie[now].next[id]){
				now = trie[now].next[id];
			}else{
				trie[now].next[id] = size++;
				now = trie[now].next[id];
			}
		}
		trie[now].cnt = no;
	}

	void build(){
		trie[0].fail = -1;
		q.push(0);
		while(q.size()){
			int u = q.front();
			q.pop();
			for(int i = 0;i<130;i++){
				if(trie[u].next[i]){
					if(u==0)	trie[trie[u].next[i]].fail = 0;
					else{
						int v = trie[u].fail;
						while(v!=-1){
							if(trie[v].next[i]){
								trie[trie[u].next[i]].fail = trie[v].next[i];
								break;
							}
							v = trie[v].fail;
						}
						if(v == -1){
							trie[trie[u].next[i]].fail = 0;
						}
					}
					q.push(trie[u].next[i]);
				}
			}
		}
	}

	int Get(int u){
		int res = 0;
		while(u){
			res += trie[u].cnt;
			trie[u].cnt = 0;
			u = trie[u].fail;
		}
		return res;
	}

	void match(char *s){
		tmp.clear();
		int res = 0,now = 0;
		int len = strlen(s);
		for(int i = 0;i<len;i++){
			int id = s[i];
			if(trie[now].next[id]){
				now = trie[now].next[id];
			}else{
				int p = now;
				while(p!=-1 && trie[p].next[id] == 0){
					p = trie[p].fail;
				}
				if(p == -1){
					now = 0;
				}else{
					now = trie[p].next[id];
				}
			}
			if(trie[now].cnt){
				tmp.push_back(trie[now].cnt);
			}
		}
	}
}aho;

int n,m;
char s[maxn];
vector<int> ve[1010];

int main(){
	scanf("%d",&n);
	aho.init();
	for(int i = 1;i<=n;i++){
		scanf("%s",s);
		aho.insert(s,i);
	}
	aho.build();
	scanf("%d",&m);
	for(int i = 1;i<=m;i++){
		scanf("%s",s);
		aho.match(s);
		sort(tmp.begin(),tmp.end());
		ve[i] = tmp;
	}
	int tot = 0;
	for(int i = 1;i<=m;i++){
		if(ve[i].size()){
			tot++;
			printf("web %d:",i);
			for(int j = 0;j<ve[i].size();j++){
				printf(" %d",ve[i][j]);
			}
			putchar('\n');
		}
	}
	printf("total: %d\n",tot);

	return 0;
}