AC自动机(模板)

\(\color{Purple}{只是照着敲了一遍，下次再来看吧~}\)

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+9;
struct Tire{
    int fail,zi[26],end;
}AC[maxn];int tot;
void insert(string s)
{
    int len=s.length(),now=0;
    for(int i=0;i<len;i++)
    {
        int ch=s[i]-'a';
        if(!AC[now].zi[ch]) AC[now].zi[ch]=++tot;
        now=AC[now].zi[ch];
    }
    AC[now].end+=1;
}
void make_fail()
{
    queue<int>q;
    for(int i=0;i<=25;i++)
    if(AC[0].zi[i]!=0)
    {
        AC[AC[0].zi[i]].fail=0;//指向根节点
        q.push(AC[0].zi[i]); 
    }
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=0;i<=25;i++)
        {
            if(AC[u].zi[i]!=0)
            {
                AC[AC[u].zi[i]].fail=AC[AC[u].fail].zi[i];
                //子节点的fail指针指向父亲fail指针指向的节点的相同子节点
                q.push(AC[u].zi[i]); 
            }
            else    AC[u].zi[i]=AC[AC[u].fail].zi[i]; 
            //没有子节点，创造子节点指向当前节点fail指针的节点 
        }
    }
}
int AC_automation(string s)//开始AC辣~ 
{
    int len=s.length(),now=0,ans=0;
    for(int i=0;i<len;i++)
    {
        now=AC[now].zi[s[i]-'a'];//向下一层冲!
        for(int j=now;j&&AC[j].end!=-1;j=AC[j].fail)
        {
            ans+=AC[j].end;
            AC[j].end=-1;
        }
    }
    return ans;
}
int main()
{
    int n;string s;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>s;
        insert(s);
    }
    AC[0].fail=0;//结束标志
    make_fail();
    cin>>s;//模式串读入
    cout<<AC_automation(s);
    return 0;//谢幕! 
}