模板 AC自动机

模板 AC自动机

code：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;

const int wx=2000017;

inline int read(){
int sum=0,f=1; char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
return sum*f;
}

struct AC_automation{
int tr[wx][26];
int val[wx],fail[wx];
int cnt;

void insert(char * s){
int p=0;
for(int i=0;s[i];i++){
int k=s[i]-'a';
if(!tr[p][k])tr[p][k]=++cnt;
p=tr[p][k];
}
val[p]++;
}

void build(){
queue<int > q; memset(fail,0,sizeof fail);
for(int i=0;i<26;i++)if(tr[0][i])q.push(tr[0][i]);
while(q.size()){
int p=q.front(); q.pop();
for(int i=0;i<26;i++){
if(tr[p][i]){
fail[tr[p][i]]=tr[fail[p]][i];
q.push(tr[p][i]);
}
else tr[p][i]=tr[fail[p]][i];
}
}
}

int query(char * s){
int p=0,re=0;
for(int i=0;s[i];i++){
p=tr[p][s[i]-'a'];
for(int j=p;j&&~val[j];j=fail[j])if(val[j]!=-1)re+=val[j],val[j]=-1;
}
return re;
}

}ac;

int n;
char c[wx];

int main(){
n=read();
for(int i=1;i<=n;i++){
scanf("%s",c);
ac.insert(c);
}
ac.build();
scanf("%s",c);
printf("%d\n",ac.query(c));
return 0;
}