hdu 2896 病毒侵袭 （ ac 自动机 ）

Problem Description 当太阳的光辉逐渐被月亮遮蔽，世界失去了光明，大地迎来最黑暗的时刻。。。。在这样的时刻，人们却异常兴奋——我们能在有生之年看到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   1   开map 容易爆空间

#include <iostream> #include <vector> #include <list> #include <map> #include <set> #include <deque> #include <queue> #include <stack> #include <bitset> #include <algorithm> #include <functional> #include <numeric> #include <utility> #include <sstream> #include <iomanip> #include <cstdio> #include <cmath> #include <cstdlib> #include <cctype> #include <string> #include <cstring> #include <cmath> #include <ctime> #define MAXN 510005 #define MAXM 164444 #define INF 100000000 #define eps 1e-7 #define L(X) X<<1 #define R(X) X<<1|1 using namespace std; bool used[151000]; bool flag; int T,n,m; struct Trie { int next[150000][130],fail[150010],ed[150000],id[150010]; int root,L; int newnode() { for(int i=0; i<130; i++) next[L][i]=-1; ed[L++]=0; id[L-1]=0; return L-1; } void init() { L=0; root=newnode(); } void insert(char buf[],int idd) { int len = strlen(buf); int now = root; for(int i=0; i<len; i++) { if(next[now][buf[i]]==-1) next[now][buf[i]]=newnode(); now = next[now][buf[i]]; } ed[now]++; id[now]=idd; } void build() { queue<int> Q; fail[root] = root; for(int i=0; i<130; i++) { if(next[root][i]==-1) { next[root][i]=root; } else { fail[next[root][i]] = root; Q.push(next[root][i]); } } while(!Q.empty()) { int now=Q.front(); Q.pop(); for(int i=0; i<130; i++) { if(next[now][i]==-1) next[now][i] = next[fail[now]][i]; else { fail[next[now][i]] = next[fail[now]][i]; Q.push(next[now][i]); } } } } int query(char buf[],int uu,int iii) { int len = strlen(buf); int now =root; int res=0; memset(used,false,sizeof(used)); for(int i=0; i<len; i++) { now =next[now][buf[i]]; int temp = now; while(temp!=root) { res += ed[temp]; if(ed[temp]>=1) { used[id[temp]]=1; flag=true; } temp = fail[temp]; } } if(!flag) return res; printf("web %d:",iii); for(int i=1;i<=n;i++) { if(used[i]) printf(" %d",i); } printf("\n"); return res; } }; char buf[110210]; Trie ac; int main() { while(scanf("%d",&n)!=EOF) { ac.init(); for(int i=0; i<n; i++) { scanf("%s",buf); ac.insert(buf,i+1); } ac.build(); scanf("%d",&m); int tot=0; for(int i=1; i<=m; i++) { flag=false; scanf("%s",buf); ac.query(buf,0,i); if(flag) tot++; } printf("total: %d\n",tot); } }