HDU-2896-病毒侵袭(AC自动机)
链接:
https://vjudge.net/problem/HDU-2896
题意:
当太阳的光辉逐渐被月亮遮蔽,世界失去了光明,大地迎来最黑暗的时刻。。。。在这样的时刻,人们却异常兴奋——我们能在有生之年看到500年一遇的世界奇观,那是多么幸福的事儿啊~~
但网路上总有那么些网站,开始借着民众的好奇心,打着介绍日食的旗号,大肆传播病毒。小t不幸成为受害者之一。小t如此生气,他决定要把世界上所有带病毒的网站都找出来。当然,谁都知道这是不可能的。小t却执意要完成这不能的任务,他说:“子子孙孙无穷匮也!”(愚公后继有人了)。
万事开头难,小t收集了好多病毒的特征码,又收集了一批诡异网站的源码,他想知道这些网站中哪些是有病毒的,又是带了怎样的病毒呢?顺便还想知道他到底收集了多少带病毒的网站。这时候他却不知道何从下手了。所以想请大家帮帮忙。小t又是个急性子哦,所以解决问题越快越好哦~~
思路:
AC自动机模板, 记录结束的标号.
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
//#include <memory.h>
#include <queue>
#include <set>
#include <map>
#include <algorithm>
#include <math.h>
#include <stack>
#include <string>
#include <assert.h>
#include <iomanip>
#include <iostream>
#include <sstream>
#define MINF 0x3f3f3f3f
using namespace std;
typedef long long LL;
const int MAXN = 2e6+10;
struct AcTree
{
int Next[128];
int fail;
int end;
void Init()
{
memset(Next, 0, sizeof(Next));
end = 0;
fail = 0;
}
}tree[MAXN];
char s[MAXN];
set<int> Vec[1010];
int n, m, cnt;
void Insert(char *s, int id)
{
int len = strlen(s);
int pos = 0;
for (int i = 0;i < len;i++)
{
if (tree[pos].Next[s[i]] == 0)
tree[pos].Next[s[i]] = ++cnt, tree[cnt].Init();
pos = tree[pos].Next[s[i]];
}
tree[pos].end = id;
}
void BuildAC()
{
queue<int> que;
for (int i = 0;i < 128;i++)
{
if (tree[0].Next[i] != 0)
{
tree[tree[0].Next[i]].fail = 0;
que.push(tree[0].Next[i]);
}
}
while (!que.empty())
{
int u = que.front();
que.pop();
for (int i = 0;i < 128;i++)
{
if (tree[u].Next[i] != 0)
{
tree[tree[u].Next[i]].fail = tree[tree[u].fail].Next[i];
que.push(tree[u].Next[i]);
}
else
tree[u].Next[i] = tree[tree[u].fail].Next[i];
}
}
}
void Query(char *s, set<int> &re)
{
int len = strlen(s);
int pos = 0;
for (int i = 0;i < len;i++)
{
pos = tree[pos].Next[s[i]];
for (int t = pos;t != 0 && tree[t].end != 0;t = tree[t].fail)
re.insert(tree[t].end);
}
}
int main()
{
while(~scanf("%d", &n))
{
cnt = 0;
tree[0].Init();
for (int i = 1; i <= n; i++)
{
scanf("%s", s);
Insert(s, i);
}
BuildAC();
scanf("%d", &m);
for (int i = 1;i <= m;i++)
Vec[i].clear();
int res = 0;
for (int i = 1; i <= m; i++)
{
scanf("%s", s);
Query(s, Vec[i]);
}
for (int i = 1; i <= m; i++)
{
if (Vec[i].size() > 0)
{
printf("web %d:", i);
for (auto x: Vec[i])
printf(" %d", x);
puts("");
res++;
}
}
printf("total: %d\n", res);
}
return 0;
}