问题:
当太阳的光辉逐渐被月亮遮蔽,世界失去了光明,大地迎来最黑暗的时刻。。。。在这样的时刻,人们却异常兴奋——我们能在有生之年看到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
思路:AC自动机的模板题
代码:
#include<stdio.h> #include<string.h> #include<queue> #define N 60010 #include<algorithm> using namespace std; int tree[60010][129],book[N]; int f[N],tag[10010]; char s[N]; int main() { int n,cnt=1,ans=1; scanf("%d",&n); memset(book,0,sizeof(book)); while(n--)//字典树的模板 { scanf("%s",s); int u=0; for(int i=0; s[i]!='\0'; i++) { int k=s[i]; if(tree[u][k]==0) tree[u][k]=cnt++; u=tree[u][k]; } book[u]=ans++;//记录第几个病毒 } queue<int>q;//把第一层的字典树放进队列中,类似于kmp的next[0]=0,只是把一个数组变成0而已; for(int i=0; i<128; i++)//第一层进队列 if(tree[0][i]) { f[tree[0][i]]=0; q.push(tree[0][i]); } while(!q.empty())//kmp的next数组遍历 { int v=q.front(); q.pop(); for(int i=0; i<128; i++) { if(tree[v][i])//如果字母v与字母i相连,则更新next值,否则返回上一层的值。 { f[tree[v][i]]=tree[f[v]][i];//类似于kmp的next[++i]=++j。 q.push(tree[v][i]);//进队列 } else tree[v][i]=tree[f[v]][i];//类似于kmp的j=next[j]。 } } int m,b[N],t,l=0; scanf("%d",&m); for(int j=1; j<=m; j++) { t=0; int flag=0,u=0; memset(tag,0,sizeof(tag)); tag[0]=1; scanf("%s",s); for(int i=0; s[i]!='\0'; i++)//类似于kmp的查询 { int w=s[i]; int v=tree[u][w]; while(v>0) { if(tag[book[v]]==0) { flag=1; b[t++]=book[v]; tag[book[v]]=1; } v=f[v]; } u=tree[u][w];//有点难懂,u走着走着会变成零 } sort(b,b+t); if(flag==1) { l++; printf("web %d:",j); for(int i=0; i<t; i++) printf(" %d",b[i]); if(j!=m) printf("\n"); } } printf("total: %d\n",l); }