HDU 2222 Keywords Search （AC自动机）

题意：

　　给一堆敏感单词，再给一篇文章，求此文章中出现几次敏感词。（多模式串匹配）

思路：

　　现在是多个模式串来匹配一篇文章，文章一般较长，如果每个串都来一次KMP的话，复杂度还是很高的，不如将文章拿来匹配这些敏感词，主要就是对这些敏感词进行处理，使得形成一棵类似于Trie树。

　　纯AC自动机的题目。以关键字建立trie树，设置好fail指针，就可以进行求出现次数了。 有几点需要注意：

　　（1）fail[t]表示的是一个点u，从root->u经过的串是root->t经过的串的后缀，比如u="asd"，t="qasd"，fail[t]=u，那么后缀是完全一样的。

　　（2）query时要注意统计全部符合条件的串，由于只是在每个模式串的最后一个节点设置tag，所以如果只顺着文章essay来走的话，可能漏统计了很多。比如有串"asd"和"sd"，而essay="asd"，那么你统计到的只是1个串，是错误的。正确做法是，根据essay串，每走到一个点t时（非root），顺着该点的fail[t]点，再fail[fail[t]]...逐个点统计tag，直到fail[x]=root为止。

#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <algorithm>
#include <vector>
#include <iostream>
#define pii pair<int,int>
#define INF 0x7f3f3f3f
#define LL long long
#define ULL unsigned long long
using namespace std;
const double PI  = acos(-1.0);
const int N=1000100;

char essay[N];

struct Trie
{
    static const int NN=5e5+10;    //节点数
    int next[NN][26], fail[NN], tag[NN];    //tag标记每个模式串尾节点
    int root, node_cnt;
    int newnode()    //创建点
    {
        for(int i=0; i<26; i++)
            next[node_cnt][i]=-1;
        tag[node_cnt]=0;    //刚创建时，默认非叶子
        return node_cnt++;
    }
    void init()
    {
        node_cnt=0;
        root=newnode(); //root是虚拟点
    }
    void insert(char s[])
    {
        int len=strlen(s);
        int now=root;
        for(int i=0; i<len; i++)
        {
            if(next[now][s[i]-'a']==-1)
                next[now][s[i]-'a']=newnode();
            now=next[now][s[i]-'a'];
        }
        tag[now]++; //尾节点:可能有多个相同模式串！
    }

    void buildAC()   //创建AC自动机：设置fail指针
    {
        queue<int> que;         //普通队列
        fail[root]=root;
        for(int i=0; i<26; i++)  //先将所有模式串头节点进队
        {
            if(next[root][i]==-1)
                next[root][i]=root;
            else
            {
                fail[next[root][i]]=root;
                que.push(next[root][i]);
            }
        }

        while(!que.empty())
        {
            int now=que.front();que.pop();
            for(int i=0; i<26; i++)
            {
                if( next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else
                {
                    fail[next[now][i]]=next[fail[now]][i];
                    que.push(next[now][i]);
                }
            }
        }
    }

    int query(char s[])
    {
        int len=strlen(s);
        int now=root;
        int ans=0;
        for(int i=0; i<len; i++)
        {
            now=next[now][s[i]-'a'];    //取next
            int tmp=now;
            while( tmp!=root )    //顺着走完它，继续统计
            {
                ans+=tag[tmp];
                tag[tmp]=0;       //统计过的置0
                tmp=fail[tmp];
            }
        }
        return ans;
    }
}AC;

int main()
{
    //freopen("input.txt","r",stdin);
    int t, n;cin>>t;
    while(t--)
    {
        scanf("%d",&n);
        AC.init();
        for(int i=0; i<n; i++)  //构建字典树
        {
            scanf("%s",essay);
            AC.insert(essay);
        }
        AC.buildAC();
        scanf("%s",essay);
        printf("%d
", AC.query(essay));
    }

    return 0;
}