本文移植自个人的原博客(原文

题目链接

HDU 2896

题目描述

当太阳的光辉逐渐被月亮遮蔽,世界失去了光明,大地迎来最黑暗的时刻。。。。在这样的时刻,人们却异常兴奋——我们能在有生之年看到500年一遇的世界奇观,那是多么幸福的事儿啊~~

但网路上总有那么些网站,开始借着民众的好奇心,打着介绍日食的旗号,大肆传播病毒。小t不幸成为受害者之一。小t如此生气,他决定要把世界上所有带病毒的网站都找出来。当然,谁都知道这是不可能的。小t却执意要完成这不能的任务,他说:“子子孙孙无穷匮也!”(愚公后继有人了)。

万事开头难,小t收集了好多病毒的特征码,又收集了一批诡异网站的源码,他想知道这些网站中哪些是有病毒的,又是带了怎样的病毒呢?顺便还想知道他到底收集了多少带病毒的网站。这时候他却不知道何从下手了。所以想请大家帮帮忙。小t又是个急性子哦,所以解决问题越快越好哦~~

输入输出格式

输入格式

第一行,一个整数$N(1<=N<=500)$,表示病毒特征码的个数。

接下来$N$行,每行表示一个病毒特征码,特征码字符串长度在$20—200$之间。

每个病毒都有一个编号,依此为$1 \sim N$。

不同编号的病毒特征码不会相同。

在这之后一行,有一个整数$M(1<=M<=1000)$,表示网站数。

接下来$M$行,每行表示一个网站源码,源码字符串长度在$7000—10000$之间。

每个网站都有一个编号,依此为$1 \sim M$。

以上字符串中字符都是$ASCII$码可见字符(不包括回车)。

输出格式

依次按如下格式输出按网站编号从小到大输出,带病毒的网站编号和包含病毒编号,每行一个含毒网站信息。

web 网站编号: 病毒编号 病毒编号 ...

冒号后有一个空格,病毒编号按从小到大排列,两个病毒编号之间用一个空格隔开,如果一个网站包含病毒,病毒数不会超过3个。

最后一行输出统计信息,如下格式。

total: 带病毒网站数

样例

输入样例

3
aaa
bbb
ccc
2
aaabbbccc
bbaacc

输出样例

web 1: 1 2 3
total: 1

题解

AC自动机模板题。

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=501*201+5;
const int maxf = 128;
int ans[1001],ans2,tmp,tr[N][maxf],tot,tag[N],fail[N];
char s[10001];
bool use[N];
void insert(int noww)
{
int n=strlen(s),now=0;
for(int i=0;i<n;++i)
{
if(tr[now][s[i]]==-1)
{
tr[now][s[i]]=++tot;
for(int j=0;j<maxf;++j)tr[tot][j]=-1;
tag[tot]=0;
}
now=tr[now][s[i]];
}
tag[now]=noww;
}
void getfail()
{
queue<int>q;
fail[0]=0;
for(int i=0;i<maxf;++i)
if(tr[0][i]!=-1)
fail[tr[0][i]]=0,q.push(tr[0][i]);
else tr[0][i]=0;
while(!q.empty())
{
int x=q.front();q.pop();
for(int j=0;j<maxf;++j)
if(tr[x][j]!=-1)
{q.push(tr[x][j]);fail[tr[x][j]]=tr[fail[x]][j];}
else
tr[x][j]=tr[fail[x]][j];
}
}
void solve(int noww)
{
int n=strlen(s),now=0;
memset(use,0,sizeof(use));
tmp=0;
for(int i=0;i<n;++i)
{
now=tr[now][s[i]];
for(int j=now;j;j=fail[j])
if(tag[j]&&!use[tag[j]])
ans[++tmp]=tag[j],use[tag[j]]=1;
}
if(tmp)
{
ans2++;
printf("web %d:",noww);
sort(ans+1,ans+tmp+1);
for(int j=1;j<=tmp;++j)
printf(" %d",ans[j]);
printf("\n");
}
}
int main()
{
int n,m;
scanf("%d",&n);
for(int i=0;i<maxf;++i)tr[0][i]=-1;
for(int i=1;i<=n;++i)
scanf("%s",s),insert(i);
getfail();
scanf("%d",&m);
for(int i=1;i<=m;++i)
scanf("%s",s),solve(i);
printf("total: %d\n",ans2);
}