AC自动机笔记

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[],fail,sum;
     void init(){
         memset(son,,sizeof(son));
         sum=;
     }
 }a[];
 queue<int>q;
 int last[],ans=,n,tot=,len;
 char st[],sp[][];
 void init(){
     for(char i='a';i<='z';i++){
         if(a[].son[u(i)]){
             q.push(a[].son[u(i)]);
             a[a[].son[u(i)]].fail=;
         }
     }
     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=;
                 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=;
     for(int i=;i<strlen(st);i++){
         if(a[s].son[u(st[i])]==){
             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=;
     }
     while(last[now]){
         ans+=a[last[now]].sum;
         a[last[now]].sum=;
         now=last[now];
     }
 }
 void solution(int len){
     int now=;
     for(int i=;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[].init();
     a[].fail=;
     for(int i=;i<=n;i++){
         scanf("%s",sp[i]);
         ins(sp[i]);
     }
     scanf("%s",st);
     len=strlen(st);
     init();
     solution(len);
     printf("%d",ans);
     return ;
 }