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
", res);
    }

    return 0;
}