*AC自动机

今天是2017/5/22，DCDCBigBig的第六篇博文

AC自动机

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define MAX 100001
#define u(n) (n-'a')
using namespace std;
struct ac{
    int son[26],fail,sum;
    void init(){
        memset(son,0,sizeof(son));
        sum=0;
    }
}a[500001];
queue<int>q;
int last[500001],ans=0,n,tot=0,len;
char st[1000001],sp[10000][51];
void init(){
    for(char i='a';i<='z';i++){
        if(a[0].son[u(i)]){
            q.push(a[0].son[u(i)]);
            a[a[0].son[u(i)]].fail=0;
        }
    }
    while(!q.empty()){
        for(char i='a';i<='z';i++){
            if(a[q.front()].son[u(i)]){
                q.push(a[q.front()].son[u(i)]);
                int now=a[q.front()].fail;
                while(now&&!a[now].son[u(i)]) now=a[now].fail;
                if(a[now].son[u(i)]) a[q.back()].fail=a[now].son[u(i)];
                else a[q.back()].fail=0;
                if(a[a[q.back()].fail].sum){
                    last[q.back()]=a[q.back()].fail;
                }else{
                    last[q.back()]=last[a[q.back()].fail];
                }
            }
        }
        q.pop();
    }
}
void ins(char st[]){
    int s=0;
    for(int i=0;i<strlen(st);i++){
        if(a[s].son[u(st[i])]==0){
            a[s].son[u(st[i])]=++tot;
            a[tot].init();
            s=tot;
        }else{
            s=a[s].son[u(st[i])];
        }
    }
    a[s].sum++;
}
void inc(int now){
    if(a[now].sum){
        ans+=a[now].sum;
        a[now].sum=0;
    }
    while(last[now]){
        ans+=a[last[now]].sum;
        a[last[now]].sum=0;
        now=last[now];
    }
}
void solution(int len){
    int now=0;
    for(int i=0;i<=len;i++){
        if(a[now].son[u(st[i])]){
            now=a[now].son[u(st[i])];
        }else{
            while(now&&!a[now].son[u(st[i])]) now=a[now].fail;
            if(a[now].son[u(st[i])]) now=a[now].son[u(st[i])];
        }
        inc(now);
    }
}
int main(){
    scanf("%d",&n);
    a[0].init();
    a[0].fail=0;
    for(int i=1;i<=n;i++){
        scanf("%s",sp[i]);
        ins(sp[i]);
    }
    scanf("%s",st);
    len=strlen(st);
    init();
    solution(len);
    printf("%d",ans);
    return 0;
}

---

死