HDU 2222 Keywords Search （AC自动机）

题意：给你一些模式串，再给你一串匹配串，问你在匹配串中出现了多少种模式串，模式串可以相同

AC自动机：trie树上进行KMP。首先模式串建立trie树，再求得失配指针（类似next数组），其作用就是在这一位不匹配时转移到失配指针上。失配指针是转移到某个等于此位置最长后缀的位置，求法是bfs

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define eps 1E-8
/*注意可能会有输出-0.000*/
#define Sgn(x) (x<-eps? -1 :x<eps? 0:1)//x为两个浮点数差的比较,注意返回整型
#define Cvs(x) (x > 0.0 ? x+eps : x-eps)//浮点数转化
#define zero(x) (((x)>0?(x):-(x))<eps)//判断是否等于0
#define mul(a,b) (a<<b)
#define dir(a,b) (a>>b)
typedef long long ll;
typedef unsigned long long ull;
const int Inf=<<;
const double Pi=acos(-1.0);
const int Mod=1e9+;
const int Max=;
const int NumSize=;
struct node
{
    int next[NumSize],coun;
    int fail;//fail指针
    void init()
    {
        memset(next,,sizeof(next));
        coun=fail=;
    }
} ACFA[Max];
int tot;
void Init()
{
    tot=;
    ACFA[tot].init();
    return;
}
char str[Max];
void Insert(int len)//添加
{
    int now=,mpos;
    for(int i=; i<len; ++i)
    {
        mpos=str[i]-'a';
        if(!ACFA[now].next[mpos])
        {
            ACFA[now].next[mpos]=++tot;
            ACFA[tot].init();
        }
        now=ACFA[now].next[mpos];
    }
    ACFA[now].coun++;
    return;
}
int que[Max];//bfs构造fail指针
void GetFail()//构造fail指针
{
    int top=,bot=;
    int now,fail;
    for(int i=;i<NumSize;++i)//入队
        if(ACFA[].next[i])
        que[top++]=ACFA[].next[i];
    while(top!=bot)
    {
        now=que[bot++];
        for(int i=;i<NumSize;++i)
        {
            if(ACFA[now].next[i])
            {
            que[top++]=ACFA[now].next[i];
            fail=ACFA[now].fail;
            while(fail&&!ACFA[fail].next[i])//寻找失配指针位置
                fail=ACFA[fail].fail;
            if(ACFA[fail].next[i])//找到
                fail=ACFA[fail].next[i];
            ACFA[ACFA[now].next[i]].fail=fail;
            }
           else//建立trie图
            ACFA[now].next[i]=ACFA[ACFA[now].fail].next[i];
        }
    }
    return;
}
int Search(int len)
{
    int ans=;
    int mpos,nowp,now=;
    for(int i=;i<len;++i)
    {
        mpos=str[i]-'a';
        while(now>&&!ACFA[now].next[mpos])//失配后
            now=ACFA[now].fail;
        if(ACFA[now].next[mpos])//找到
        {
            now=ACFA[now].next[mpos];
            nowp=now;
            while(nowp&&ACFA[nowp].coun!=-)//-1找过
            {
                ans+=ACFA[nowp].coun;
                ACFA[nowp].coun=-;
                nowp=ACFA[nowp].fail;
            }
        }
    }
    return ans;
}
int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
        Init();
        scanf("%d",&n);
        for(int i=; i<n; ++i)
        {
            scanf("%s",str);
            Insert(strlen(str));
        }
        GetFail();
        scanf("%s",str);
        printf("%d\n",Search(strlen(str)));
    }
    return ;
}