HDU 2896 病毒侵袭

HDU 2896 病毒侵袭

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

Problem Description - 题目描述

　　当太阳的光辉逐渐被月亮遮蔽，世界失去了光明，大地迎来最黑暗的时刻。。。。在这样的时刻，人们却异常兴奋——我们能在有生之年看到500年一遇的世界奇观，那是多么幸福的事儿啊~~
　　但网路上总有那么些网站，开始借着民众的好奇心，打着介绍日食的旗号，大肆传播病毒。小t不幸成为受害者之一。小t如此生气，他决定要把世界上所有带病毒的网站都找出来。当然，谁都知道这是不可能的。小t却执意要完成这不能的任务，他说：“子子孙孙无穷匮也！”（愚公后继有人了）。
　　万事开头难，小t收集了好多病毒的特征码，又收集了一批诡异网站的源码，他想知道这些网站中哪些是有病毒的，又是带了怎样的病毒呢？顺便还想知道他到底收集了多少带病毒的网站。这时候他却不知道何从下手了。所以想请大家帮帮忙。小t又是个急性子哦，所以解决问题越快越好哦~~

 

Input - 输入	Output - 输出
第一行，一个整数N（1<=N<=500），表示病毒特征码的个数。 接下来N行，每行表示一个病毒特征码，特征码字符串长度在20—200之间。 每个病毒都有一个编号，依此为1—N。 不同编号的病毒特征码不会相同。 在这之后一行，有一个整数M（1<=M<=1000），表示网站数。 接下来M行，每行表示一个网站源码，源码字符串长度在7000—10000之间。 每个网站都有一个编号，依此为1—M。 以上字符串中字符都是ASCII码可见字符（不包括回车）。	第一行，一个整数N（1<=N<=500），表示病毒特征码的个数。 接下来N行，每行表示一个病毒特征码，特征码字符串长度在20—200之间。 每个病毒都有一个编号，依此为1—N。 不同编号的病毒特征码不会相同。 在这之后一行，有一个整数M（1<=M<=1000），表示网站数。 接下来M行，每行表示一个网站源码，源码字符串长度在7000—10000之间。 每个网站都有一个编号，依此为1—M。 以上字符串中字符都是ASCII码可见字符（不包括回车）。

Sample Input - 输入样例	Sample Output - 输出样例
3 aaa bbb ccc 2 aaabbbccc bbaacc	web 1: 1 2 3 total: 1

 【题解】

　　普通的AC自动机。多开辟一个数组保存初始状态即可，唯一感觉有点坑的就是输出total: X后还要来个回车，不然会PE。

【代码 C++】

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
int tr[100005][128], iTR, fail[100005], cnt[100005], cnt2[100005];
int opt[505], iOP;

void build(){
    int n;
    scanf("%d", &n), getchar();
    int i, j, k;
    char word[205];
    for (i = 1; i <= n; ++i){
        gets(word);
        for (j = k = 0; word[j]; ++j){
            if (!tr[k][word[j]]) tr[k][word[j]] = ++iTR;
            k = tr[k][word[j]];
        }
        cnt2[k] = i;
    }
}
void setFail(){
    std::queue<int> q;
    int i, j, now;
    for (i = 0; i < 128; ++i) if (tr[0][i]) q.push(tr[0][i]);
    while (!q.empty()){
        now = q.front(); q.pop();
        for (i = 0; i < 128; ++i){
            if (j = tr[now][i]) q.push(j), fail[j] = tr[fail[now]][i];
            else tr[now][i] = tr[fail[now]][i];
        }
    }
}
void fid(){
    memcpy(cnt, cnt2, sizeof(cnt2));
    int i, j, temp;
    char text[10005]; gets(text);
    for (i = j = iOP = 0; text[i]; ++i){
        j = tr[j][text[i]];
        for (temp = j; ~cnt[temp]; temp = fail[temp]){
            if (cnt[temp]) opt[iOP++] = cnt[temp];
            cnt[temp] = -1;
        }
    }
    std::sort(opt, opt + iOP);
}
int main(){
    build();
    setFail();
    int i = 1, j, m, total = 0;
    for (scanf("%d", &m), getchar(); i <= m; ++i){
        fid();
        if (iOP){
            ++total; printf("web %d:", i);
            for (j = 0; j < iOP; ++j) printf(" %d", opt[j]);
            puts("");
        }
    }
    printf("total: %d
", total);
    return 0;
}