hdu 2896 AC自动机 保存编号

题意：

天朝语题。

Input

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

Output

依次按如下格式输出按网站编号从小到大输出，带病毒的网站编号和包含病毒编号，每行一个含毒网站信息。
web 网站编号: 病毒编号 病毒编号 …
冒号后有一个空格，病毒编号按从小到大排列，两个病毒编号之间用一个空格隔开，如果一个网站包含病毒，病毒数不会超过3个。
最后一行输出统计信息，如下格式
total: 带病毒网站数
冒号后有一个空格。

解析：

首先，用end[i]表示以id结尾的病毒。

然后我本来用vector来保存病毒的，TLE了。

string也会TLE。

然后直接用str[i]来当作下标，否则RE。

坐标要开到128。

next end fail 数组不能开太小，否则MLE。 -  -  MLE。

然后最后结果还要排个序的，不然wa。

代码：

#pragma comment(linker, "/STACK:1677721600")
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
#include <cassert>
#include <iostream>
#include <algorithm>
#define pb push_back
#define mp make_pair
#define LL long long
#define lson lo,mi,rt<<1
#define rson mi+1,hi,rt<<1|1
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a,b) memset(a,b,sizeof(a))
#define FIN freopen("in.txt", "r", stdin)
#define FOUT freopen("out.txt", "w", stdout)

using namespace std;
const int mod = 1e9 + 7;
const double eps = 1e-8;
const double ee = exp(1.0);
const int inf = 0x3f3f3f3f;
const int maxn = 500 + 10;
const int maxm = 1000 + 10;
const double pi = acos(-1.0);
const LL iinf = 0x3f3f3f3f3f3f3f3f;

const int dictSize = 128;

//vector<int> web[maxm];
int web[maxm][maxn];
int cnt[maxm];
int webNum;

struct Trie
{
    int next[maxn * dictSize][dictSize];   //next[i][j]保存节点i的那个编号为j的节点(小写字母按字典序编号为0-(a),1-(b),2-(c),...)
    int fail[maxn * dictSize];             //后缀链接 fail[j]表示节点j沿着失配指针往回走时 遇到的下一个单词节点编号
    int end[maxn * dictSize];              ///保存当前病毒编号
    int rt;                                //根
    int nodeNum;                           //节点个数

    void init()
    {
        nodeNum = 0;
        rt = newNode();
    }

    int newNode()
    {
        for (int i = 0; i < dictSize; i++)
            next[nodeNum][i] = -1;
        end[nodeNum++] = 0;
        return nodeNum - 1;
    }

    //将字符串str加入Trie前缀树中
    void insert(char str[], int id)
    {
        int len = strlen(str);
        int now = rt;
        for (int i = 0; i < len; i++)
        {
            if (next[now][str[i]] == -1)
            {
                next[now][str[i]] = newNode();
            }
            now = next[now][str[i]];
        }
        end[now] = id;
    }

    //建立后缀链接
    void build()
    {
        queue<int> q;
        fail[rt] = rt;
        for (int i = 0; i < dictSize; i++)
        {
            if (next[rt][i] == -1)
            {
                next[rt][i] = rt;
            }
            else
            {
                fail[next[rt][i]] = rt;
                q.push(next[rt][i]);
            }
        }
        while (!q.empty())
        {
            int now = q.front();
            q.pop();
            for (int i = 0; i < dictSize; i++)
            {
                if (next[now][i] == -1)
                {
                    next[now][i] = next[fail[now]][i];
                }
                else
                {
                    fail[next[now][i]] = next[fail[now]][i];
                    q.push(next[now][i]);
                }
            }
        }
    }

    //询问子串们在串str中出现的次数
    void query(char str[], int id)
    {
        int len = strlen(str);
        int now = rt;
        for (int i = 0; i < len; i++)
        {
            now = next[now][str[i]];
            int temp = now;
            while (temp != rt)
            {
                if (end[temp] != 0)
                    web[id][cnt[id]++] = end[temp];
                temp = fail[temp];
            }
        }
    }

    void debug()
    {
        for (int i = 0; i < nodeNum; i++)
        {
            printf("id = %3d,fail = %3d,end = %3d,chi = [",i,fail[i],end[i]);
            for(int j = 0;j < 26;j++)
                printf("%2d",next[i][j]);
            printf("]\n");
        }
    }
} ac;

char str[10000 + 10];

int main()
{
#ifdef LOCAL
    FIN;
#endif // LOCAL
    int n, m;
    while (~scanf("%d", &n))
    {
        ac.init();
        for (int i = 1; i <= n; i++)
        {
            scanf("%s", str);
            ac.insert(str, i);
        }
        ac.build();

        scanf("%d", &m);
        mem0(web);
        mem0(cnt);
        int total = 0;
        for (int i = 1; i <= m; i++)
        {
            scanf("%s", str);
            ac.query(str, i);
            int sz = cnt[i];
            if (sz)
            {
                total++;
                printf("web %d:", i);
                sort(web[i], web[i] + sz);
                for (int j = 0; j < sz; j++)
                {
                    printf(" %d", web[i][j]);
                }
                puts("");
            }
        }
        printf("total: %d\n", total);
    }
    return 0;
}