BZOJ 3940 AC自动机

思路:
需要维护一个栈的AC自动机…….
要求出来 最后的栈顶是在自动机上的哪个节点。

if(!ac.ch[st[tp-1]][a[i]-'a']) st[tp]=ac.ch[ac.f[st[tp-1]]][a[i]-'a'];
else st[tp]=ac.ch[st[tp-1]][a[i]-'a'];

如果ch[][] 到不了根 就要走到fail

//By SiriusRen
#include <queue>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 500050
#define M 26
int n,st[100050],tp=0;
char a[100050],jy[100500];
struct AC_Automata{
    int sz,ch[N][M],f[N],len[N];
    void insert(char s[],int num){
        int u=0,i;
        for(i=0;s[i];i++){
            if(!ch[u][s[i]-'a'])ch[u][s[i]-'a']=++sz;
            u=ch[u][s[i]-'a'];
        }
        len[u]=i;
    }
    void build(){
        f[0]=500000;
        queue<int>q;q.push(0);
        while(!q.empty()){
            int r=q.front();q.pop();
            for(int i=0;i<M;i++){
                int u=ch[r][i];
                if(!u)ch[r][i]=ch[f[r]][i];
                else f[u]=ch[f[r]][i],len[u]=max(len[u],len[f[u]]),q.push(u);
            }
        }
    }
}ac;
int main(){
    scanf("%s%d",a,&n);
    for(int i=1;i<=n;i++)
        scanf("%s",jy),ac.insert(jy,i);
    ac.build();
    for(int i=0;a[i];i++){
        jy[++tp]=a[i];
        if(!ac.ch[st[tp-1]][a[i]-'a']) st[tp]=ac.ch[ac.f[st[tp-1]]][a[i]-'a'];
        else st[tp]=ac.ch[st[tp-1]][a[i]-'a'];
        tp-=ac.len[st[tp]];
    }
    jy[tp+1]=0;
    printf("%s",jy+1);
}
posted @ 2016-12-09 16:32  SiriusRen  阅读(191)  评论(0编辑  收藏  举报