AC自动机模板（hdu2222）

刚刚学习了AC 自动机，先记录一个数组写法的模板。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define Max 26
#define N 1000005

struct node
{
    int next[Max];//每一个节点可以扩展到的字母
    int fail;//每一个节点的失配指针
    int count;
    void init()//构造
    {
        memset(next,-1,sizeof(next));
        fail=0;
        count=0;
    }
}s[N];
int num;//记录节点编号
char str[55];//模板串，单词
char ss[N];//母串
int q[N],tail,head;//队列相关信息
void cas_init()//在整个程序前构造root
{
    s[0].init();//初始化头结点
    num=1;//当前的节点数量为1
}
void insert()
{
    int len=strlen(str);
    int p=0,i,j;
    for(i=0;i<len;i++)
    {
        j=str[i]-'a';//求出字母在next中的编号
        if(s[p].next[j]==-1)//如果为空则构造新的，否则顺着上次的开始往下构造
        {
            s[num].init();//初始化当前节点
            s[p].next[j]=num++;//连向当前节点，并是num++来扩充节点
        }
        p=s[p].next[j];//向下遍历
    }
    s[p].count++;//增加离根节点这条路径
}
void build_ac()
{
    tail=head=0;//初始化队列
    int temp,p;
    for(int i=0;i<Max;i++)
    {
        if(s[0].next[i]!=-1)
        {
            q[tail++]=s[0].next[i];//和根节点相连的都入队
        }
    }
    while(tail!=head)
    {
        p=q[head++];//记录队首节点
        for(int i=0;i<Max;i++)//遍历首节点的next
        {
            if(s[p].next[i]!=-1)//如果节点next不为空
            {
                q[tail++]=s[p].next[i];//将儿子节点入队列
                temp=s[p].fail;//记录节点的失配指针指向
                while(temp>0&&s[temp].next[i]==-1)//当失配指针不为root时一直循环找到一的儿子节点不为空或到了root
                temp=s[temp].fail;
                if(s[temp].next[i]!=-1)//如果当前节点有儿子的话记录下来备用
                temp=s[temp].next[i];
                s[s[p].next[i]].fail=temp;//是当前节点的失配指针指向刚才记录的节点完成失配指针的构造
            }
        }
    }
}
void query()
{
    int len=strlen(ss);
    int p=0;
    int temp=0;
    int ans=0;
    for(int i=0;i<len;i++)
    {
        int id=ss[i]-'a';
        while(p>0&&s[p].next[id]==-1)//当前p指针不是root和找不到儿子时，一直找下去（及kmp的while）
        p=s[p].fail;//一直寻找失配指针
        if(s[p].next[id]!=-1)//找到适合的失配指针
        {
            p=s[p].next[id];//指向这个儿子节点，更新p的值进行下次匹配
            temp=p;
            while(temp>0&&s[temp].count!=-1)//temp > 0表示还没到root，count != -1表示指针前还有单词
            {
                ans+=s[temp].count;//加上有的单词个数
                s[temp].count=-1;//不重复计算
                temp=s[temp].fail;//一直寻找失配指针
            }
        }
    }
    printf("%d\n",ans);
}
int main()
{
    int n;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        cas_init();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str);
            insert();
        }
        build_ac();
        scanf("%s",ss);
        query();
    }
    return 0;
}