[HDOJ2222]Keywords Search（AC自动机）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2222

题意：求字典中所有模式串各出现了多少次。

建立AC自动机，查询的时候首先跳过所有cur->next[index]为NULL的情况，再跳过cur是root的情况。因为root不存任何信息。

过滤完上述情况，那么当前匹配串上的index位置必然在trie上出现过，那么就沿着当前点的fail一路找上去。因为trie某节点的fail指针要么指向root，要么指向与当前字符相同的节点。接着记下每个点的cnt值，不要忘记置零防止重复统计。

时隔多年（什么多年啊喂）重拾AC自动机，终于彻彻底底弄明白了。感恩的心。

#include <bits/stdc++.h>
using namespace std;

inline int GetId(char c) {
    return c - 'a';
}
typedef class Node {
public:
    Node *next[26];
    Node *fail;
    int cnt;
    Node() {
        memset(next, 0, sizeof(next));
        fail = NULL;
        cnt = 0;
    }
}Node;
class AC_Automation {
public:
    Node *root;
    queue <Node*> q;
    AC_Automation() {
        root = new Node;
        while(!q.empty()) q.pop();
    }
    void insert(string s) {
        Node *cur = root;
        int len = s.length();
        for(int i = 0; i < len; i++) {
            int index = GetId(s[i]);
            if(cur->next[index] == NULL) {
                cur->next[index] = new Node;
            }
            cur = cur->next[index];
        }
        cur->cnt++;
    }
    void BuildAC() {
        Node *cur,*tmp;
        q.push(root);
        while(!q.empty()) {
            cur = q.front();
            q.pop();
            for(int i = 0; i < 26; i++) {
                if(cur->next[i]) {
                    if(cur == root) {
                        cur->next[i]->fail = root;
                    }
                    else {
                        tmp = cur->fail;
                        while(tmp != NULL) {
                            if(tmp->next[i]) {
                                cur->next[i]->fail = tmp->next[i];
                                break;
                            }
                            tmp = tmp->fail;
                        } 
                        if(tmp == NULL) {
                            cur->next[i]->fail = root;
                        }
                    }
                    q.push(cur->next[i]);
                }
            }
        }
    }
    int query(string s) {
        Node *cur = root,*tmp;
        int len = s.length();
        int ret = 0;
        for(int i = 0; i < len; i++) {
            int index = GetId(s[i]);
            while(cur->next[index] == NULL && cur != root) {
                cur = cur->fail;
            }
            cur = cur->next[index];
            if(cur == NULL) {
                cur = root;
                continue;
            }
            tmp = cur;
            while(tmp != root) {
                ret += tmp->cnt;
                tmp->cnt = 0;
                tmp = tmp->fail;
            }
        }
        return ret;
    }
};
char pat[52], tar[2000005];
int n;

int main() {
    // freopen("in", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &n);
        AC_Automation ac;
        for(int i = 0; i < n; i++) {
            scanf("%s", pat);
            ac.insert(pat);
        }
        ac.BuildAC();
        scanf("%s", tar);
        cout << ac.query(tar) << endl;
    }
    return 0;
}