【HDU2896】病毒侵袭——AC自动机基础

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

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

Source

2009 Multi-University Training Contest 10 - Host by NIT

G++ MLE c++ AC

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

typedef struct node
{
int Id;

int last;

node *next[100];

node *fail;

node()
{
Id = 0;

last =0;

for(int i = 0;i<100;i++)
{
next[i] = NULL;
}

fail = NULL;
}
}Tree;

char str[10100];

int n,m;

int a[5],top;

void BuildTree(int num,Tree *Root)
{
int len  =strlen(str);

for(int i = 0;i<len;i++)
{
int ans = str[i]-32;

if(Root->next[ans]==NULL) Root->next[ans] = new node();

Root = Root->next[ans];
}

Root->Id = num;
}

void BFS(Tree *Root)
{
queue<Tree*>Q;

Q.push(Root);

Tree *u,*v;

while(!Q.empty())
{
u = Q.front();

Q.pop();

v  = NULL;

for(int i = 0;i<100;i++)
{
if(u->next[i] == NULL) continue;

if(u == Root)
{
u->next[i]->fail = Root;
}
else
{
v = u->fail;

while(v!=NULL)
{
if(v->next[i]==NULL)
{
v = v->fail;
}
else
{
u->next[i]->fail = v->next[i];

break;
}
}

if(v==NULL)
{
u->next[i]->fail = Root;
}
}

Q.push(u->next[i]);
}
}
}

void Query(Tree * Root,int num)
{
int len = strlen(str);

Tree * p = Root;

for(int i = 0 ;i<len;i++)
{
int ans = str[i] - 32;

while(p->next[ans]==NULL&&p!=Root) p = p->fail;

p = p->next[ans];

if(p==NULL) p =Root;

Tree *temp =  p;

while(temp !=Root &&temp->last != num)//判断之间是不是访问过
{
if(temp->Id)
{
a[top++] = temp->Id;
}

temp->last = num;

if(top>=3)//不会大于3个
{
return ;
}

temp = temp->fail;
}
}
}

int main()
{

while(~scanf("%d",&n))
{

Tree * Root;

Root = new node();

for(int i = 1;i<=n;i++)
{
scanf("%s",str);

BuildTree(i,Root);
}

BFS(Root);

int ans = 0;

scanf("%d",&m);

for(int i = 1;i<=m;i++)
{
scanf("%s",str);

top = 0;

Query(Root,i);

if(!top) continue;

printf("web %d:",i);

sort(a,a+top);

for(int j = 0;j<top;j++) printf(" %d",a[j]);

printf("\n");

if(top) ans++;
}

printf("total: %d\n",ans);
}
return 0;
}