p3808 ac自动机模板

ac自动机模板

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>

using namespace std;
const int maxn=1e6+7;
int trie[maxn][26],idx,val[maxn],fail[maxn];
char s[maxn];
int n;

void add(){
    int len=strlen(s),now=0,id;
    for(int i=0;i<len;i++){
        id=s[i]-'a';
        if(!trie[now][id])trie[now][id]=++idx;
        now=trie[now][id];
    }
    val[now]++;
}
void build(){
    queue<int>q;
    for(int i=0;i<26;i++)if(trie[0][i])q.push(trie[0][i]),fail[trie[0][i]]=0;
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=0;i<26;i++)
            if(trie[x][i])fail[trie[x][i]]=trie[fail[x]][i],q.push(trie[x][i]);
            else trie[x][i]=trie[fail[x]][i];
    }
}
int query(){
    int len=strlen(s),now=0,ans=0;
    for(int i=0;i<len;i++){
        now=trie[now][s[i]-'a'];
        for(int t=now;t&&~val[t];t=fail[t])ans+=val[t],val[t]=-1;
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    while(n--)scanf("%s",&s),add();
    scanf("%s",&s);build();
    printf("%d",query());
    return 0;
}