HDU - 2896 病毒侵袭 AC自动机

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

题意：

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

题解：有点卡内存。。

因为这个目标串只有200.所以每次找fail的时候直接找也不会超时，如果长的话，那就需要每次开个vis标记一下，对于每次查询只回溯一次就好

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int vis[510];
struct AC {
	int tree[N][130];
	int fail[N];
	int val[N];
	int tot;
	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(int x, char *str) {
		memset(vis, 0, sizeof(vis));
		int flag = 0;
		int p = 0, res = 0;
		int len = strlen(str);
		for(int i = 0; i < len; i++) {
			p = tree[p][str[i]];
			for(int j = p; j; j = fail[j]) { // 不断找fail 
				if(val[j]) {
					vis[val[j]] = 1;
					flag = 1;
				}
			}
		}
		if(!flag) return false;
		printf("web %d:", x);
		for(int i = 1; i <= 500; i++)
			if(vis[i])
				printf(" %d", i);
		printf("\n");
		return true;
	}
}A;
int n, m;
int main() {
	char s[N];
	scanf("%d", &n);
	for(int i = 1; i <= n; i++) {
		scanf("%s", s);
		A.insert(i, s);
	}	
	A.getfail();
	scanf("%d", &m);
	int res = 0;
	for(int i = 1; i <= m; i++) {
		scanf("%s", s);
		if(A.query(i, s)) res++;
	}
	printf("total: %d\n", res);
	return 0;
}

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int vis[N];
vector<int > ans;
struct AC {
	int tree[N][130];
	int fail[N];
	int val[N];
	int tot;
	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(int x, 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 && !vis[j]; j = fail[j]) { // 不断找fail 
	            if(val[j]) {
	                ans.push_back(val[j]);
	            }
	            vis[j] = 1;
	        }
	    }
	    if(ans.size() == 0) return false;
	    printf("web %d:", x);
	    sort(ans.begin(), ans.end());
	    for(int i = 0; i < ans.size(); i++)
	        printf(" %d", ans[i]);
	    printf("\n");
	    return true;
	}
}A;
int n, m;
int main() {
	char s[N];
	scanf("%d", &n);
	for(int i = 1; i <= n; i++) {
		scanf("%s", s);
		A.insert(i, s);
	}	
	A.getfail();
	scanf("%d", &m);
	int res = 0;
	for(int i = 1; i <= m; i++) {
		scanf("%s", s);
		if(A.query(i, s)) res++;
	}
	printf("total: %d\n", res);
	return 0;
}