BZOJ 3940: [Usaco2015 Feb]Censoring AC自动机_栈

Code:

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)  // ,freopen(s".out","w",stdout) 
#define maxn 200000 
using namespace std;
char str[maxn],arr[maxn]; 
int S[maxn],pos[maxn],top; 
namespace AC
{
    #define sigma 30 
    queue<int>Q; 
    int tot=0,root=0; 
    int ch[maxn][30],tag[maxn],f[maxn]; 
    void ins(char p[],int n)
    {
        int x=root; 
        for(int i=1;i<=n;++i)
        {
            int c=p[i]-'a';
            if(!ch[x][c]) ch[x][c]=++tot; 
            x=ch[x][c]; 
        } 
        tag[x]=n;  
    }
    void getfail()
    { 
        for(int i=0;i<sigma;++i) if(ch[root][i]) Q.push(ch[root][i]);    
        while(!Q.empty())
        {
            int u=Q.front();Q.pop(); 
            for(int i=0;i<sigma;++i)
            {
                int q=ch[u][i];
                if(!q) { ch[u][i]=ch[f[u]][i]; continue; }
                Q.push(q); 
                f[q]=ch[f[u]][i];             
            }
        }
    }     
}; 
int main()
{
    // setIO("input");
    int len,m; 
    scanf("%s",str+1),len=strlen(str+1); 
    scanf("%d",&m);  
    for(int i=1,a;i<=m;++i) 
    {
        scanf("%s",arr+1), a=strlen(arr+1), AC::ins(arr,a);
    }
    AC::getfail(); 
    int tr=0; 
    for(int i=1;i<=len;++i)
    { 
        int c=str[i]-'a'; 
        tr=AC::ch[tr][c];   
        pos[i]=tr;             
        S[++top]=i;   
        if(AC::tag[tr]) top -= AC::tag[tr], tr = pos[S[top]];                  
    }        
    for(int i=1;i<=top;++i) printf("%c",str[S[i]]); 
    printf("\n"); 
    return 0; 
}