hdu2222 ac自动机模板

题意：给出一系列字符串作为关键字，再给出一个字符串作为查询串，问查询串里包含多少关键字。

字符串匹配有kmp，保存大量字符串有字典树，所以当出现大量字符串需要匹配的情况，就有了ac自动机，这名字总让人想到一种很美好的歧义。。

就是数据插入字典树，构造自动机和把字符串放进去匹配三个基本操作把。

构造自动机规则如下：

如果某个节点的父节点就是根节点，那么这个节点的失败节点也就是根节点。

如果C的子节点中，有与A相同的节点B，则A的失败节点就是B

如果C的子节点中，没有与A相同的节点，且C就是根节点，则A的失败节点就是C

如果C的子节点中，没有与A相同的节点，但C不是根节点，则令C为C的失败节点，重复

查询规则如下：

在查找时，当前处在某个节点，往下无路可走时，那么就可以沿着fail指针跳到失败节点。

如果失败节点也无路可走，则跳到失败节点的失败节点，直至有路可走或者走到根节点为止。

思路就那么回事，真要写就纠结了。。

模板代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;
const int maxn = 500010;
char str[1000010];

struct ACAutomaton{
    int next[maxn][26],fail[maxn],flag[maxn],last[maxn]；
    int num,root;
    int newnode()
    {
        memset(next[num],0,sizeof(next[num]));
        flag[num] = 0;
        return num++;//每次都加，故num为结点数
    }

    void init()
    {
        num = 0;
        root = newnode();
    }

    void insert(char *str)
    {
        int len = strlen(str);
        int now = root;
        for(int i = 0;i < len;i++)
        {
            int &tmp = next[now][str[i]-'a'];//&取地址
            if(!tmp)
                tmp = newnode();//是空的那新建结点
            now = tmp;
        }
        flag[now]++;
    }

    void getfail()
    {
        queue<int> q;
        fail[root] = root;
        for(int i = 0;i < 26;i++)
        {
            int u = next[root][i];
            if(u)//头
            {
                fail[u] = last[u] = 0;
                q.push(u);
            }
        }
        while(!q.empty())
        {
            int now = q.front();q.pop();
            for(int i = 0;i < 26;i++)
            {
                int u = next[now][i];
                if(!u)
                    //这个点的第i个儿子指向now的失败节点的第i个儿子
                    next[now][i] = next[fail[now]][i];
                else
                {
                    fail[u] = next[fail[now]][i];//u的失败节点指向fail[now]的第i个儿子
                    last[u] = flag[fail[u]] ? fail[u]:last[fail[u]];//后缀链接
                    q.push(u);
                }
            }
        }
    }

    int query(char *str){
        int len = strlen(str);
        int now = root;
        int ret = 0;
        for(int i = 0;i < len;i++)
        {
            now = next[now][str[i]-'a'];
            int tmp = now;
            while(tmp != root && flag[tmp])
            {
                ret += flag[tmp];
                flag[tmp] = 0;//防止重复统计
                tmp = last[tmp];
            }
        }
        return ret;
    }
}ac;

int main(){
    int k,n;
    scanf("%d",&k);
    while(k--)
    {
        ac.init();
        scanf("%d",&n);
        for(int i = 0;i < n;i++)
        {
            scanf("%s",str);
            ac.insert(str);
        }
        ac.getfail();//类似kmp里的next
        scanf("%s",str);
        printf("%d\n",ac.query(str));
    }
    return 0;
}