洛谷p3121 ac自动机

ac自动机匹配时，建立两个栈，其中一个栈存储答案，另一个栈存储失配指针。

如果在某点找到我们要删除的单词（全部），那么我们要将匹配指针前移至此单词前面那个字母，即将我们的指针前移长度单位。

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <queue>
#include <cstring>
#include <algorithm>
#include <string>
using namespace std;
//由于题目中子串美丽度计算时不可重复，因此记录每个字符串长度，

struct Tire{
    int nxt[1000100][26],fail[1000100],ed[1000100],los[1000100];
    int cnt,rt;
    int ans[1000100];
    int newnode()
    {
        for(int i=0;i<26;i++)
        {
            nxt[cnt][i]=-1;
        }
        ed[cnt++]=-1;
        return cnt-1;
    }
    void init()
    {
        cnt=0;
        rt=newnode();
    }
    void ins(string s,int id)
    {
        int len=s.size();
        int now=rt;
        for(int i=0;i<len;i++)
        {
            if(nxt[now][s[i]-'a']==-1)
            {
                nxt[now][s[i]-'a']=newnode();
            }
            now=nxt[now][s[i]-'a'];
        }
        ed[now]=id;
    }
    void build()
    {
        queue<int>q;
        fail[rt]=rt;
        for(int i=0;i<26;i++)
        {
            if(nxt[rt][i]==-1)
            {
                nxt[rt][i]=rt;
            }
            else
            {
                fail[nxt[rt][i]]=rt;
                q.push(nxt[rt][i]);
            }
        }
        while(!q.empty())
        {
            int now=q.front();
            q.pop();
            for(int i=0;i<26;i++)
            {
                if(nxt[now][i]==-1)
                {
                    nxt[now][i]=nxt[fail[now]][i];
                }
                else
                {
                    fail[nxt[now][i]]=nxt[fail[now]][i];
                    q.push(nxt[now][i]);
                }
            }
        }
    }
    void query(string s)
    {
        int now=rt,top=0;
       for(int i=0;i<s.size();i++)
       {
           los[i]=now=nxt[now][s[i]-'a'];
           ans[++top]=i;
           if(ed[now]!=-1)
           {
               top-=ed[now];
               now=los[ans[top]];
           }
       }
       for(int i=1;i<=top;i++)
        printf("%c",s[ans[i]]);
       puts("");
    }
};
Tire ac;
string s,s1;

int main()
{
        int n;
        cin>>s1;
        scanf("%d",&n);
        ac.init();
        for(int i=0;i<n;i++)
        {
             cin>>s;
             ac.ins(s,s.size());
        }
        ac.build();
        ac.query(s1);
    return 0;
}