【AC自动机】HDU - 2896 病毒侵袭（字典树应该开多大的数组？？？）

 HDU - 2896 病毒侵袭

• 题意：N个模式串，M个文本串。对于每个文本串，输出其中出现的模式串的序号。最后一行输出包含模式串的文本串的个数。字符来自于可打印ascii码。

---

不用说肯定是AC自动机的模板的嘛，但是怎么说，我竟然自己把自己卡在了数组大小上。

字典树结点数到底应该开多大？本来我是开了N*len*100(因为ascii码可打印的个数是90+)。然后编译都没过，确实叭，数组太大了。为了看有没有其他错误的嘛，于是开了N*len运行代码。样例过了之后交了一发T了。于是把Fail树构建给删了（因为没有让求每个模式串的出现次数，所以没必要的，不知道写的时候咋想的，反正写上了，可能顺手了……），删了之后又交了一发，竟然过了！！？？

于是开始百度字典树应该开多大数组？然后大部分就是说的是N*len=模式串个数*模式串长度。想想确实好像也是。但是不知道为什么就是觉得……有点诡异。这个AC有点让我不知所措……

---

ps: 昨天上课的英语老师竟然也是个程序员。他问我coding是喜欢还是迫不得已。我说喜欢。他说他是工作迫不得已。233333333333

---

AC CODE

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxS = 10004;
const int maxN = 100005;
int total;
struct AC_automat{
    int trie[maxN][100], tot;
    int fail[maxN];
    int times[maxN];
    int endd[maxS];

    void Clear(int rt)
    {
        for(int i = 0; i < 100; ++ i )
            trie[rt][i] = 0;
        fail[rt] = 0;
    }

    void init()
    {
        Clear(0);
        tot = 0;
        total = 0;
    }

    void Insert(char *s, int p)
    {
        int rt = 0;
        for(int i = 0; s[i]; ++ i )
        {
            int id = s[i] - 32;
            if(!trie[rt][id]) { trie[rt][id] = ++ tot; Clear(tot); }
            rt = trie[rt][id];
        }
        endd[p] = rt;
    }

    void build()
    {
        queue<int>q;
        for(int i = 0; i < 100; ++ i) if(trie[0][i]) q.push(trie[0][i]);
        while(!q.empty())
        {
            int rt = q.front(); q.pop();
            for(int i = 0; i < 100; ++ i )
            {
                if(trie[rt][i])
                {
                    fail[trie[rt][i]] = trie[fail[rt]][i];
                    q.push(trie[rt][i]);
                } else trie[rt][i] = trie[fail[rt]][i];
            }
        }
    }

    void getTimes(char *t)
    {
        memset(times, 0, sizeof(times));
        int rt = 0;
        for(int i = 0; t[i]; ++ i )
            ++ times[rt = trie[rt][t[i] - 32]];
    }

    void print(int n, int id)
    {
        vector<int>ans;
        for(int i = 1; i <= n; ++ i)
        {
            if(times[endd[i]])
                ans.push_back(i);
        }
        if(!ans.empty())
        {
            ++ total;
            printf("web %d: ", id);
            int siz = ans.size();
            for(int i = 0; i < siz; ++ i)
                printf("%d%c", ans[i], " \n"[i == siz - 1]);
        }
    }

}ac;

int n, m;
char s[maxS];

int main()
{
    scanf("%d", &n); ac.init();
    getchar();
    for(int i = 1; i <= n; ++ i)
    {
        gets(s);
        ac.Insert(s, i);
    }
    ac.build();
    scanf("%d", &m);
    getchar();
    for(int i = 1; i <= m; ++ i )
    {
        gets(s);
        ac.getTimes(s);
        ac.print(n, i);
    }
    printf("total: %d\n", total);
}