POJ2222 Keywords Search AC自动机模板

http://acm.hdu.edu.cn/showproblem.php?pid=2222

题意:给出一些单词，求多少个单词在字符串中出现过（单词表单词可能有相同的,这些相同的单词视为不同的分别计数）（如果单词x在字符串中出现一次而在单词表中有两个则ans+2,在字符串中出现两次而单词表中有一个则ans+1）

AC自动机模板题，之前学了总觉得不扎实，现在有底了现在终于可以去敲心头大恨，兴奋！！！

代码

#include<cstdio>
#include<cstring>
const int maxn=10010;
const double eps=1e-8;
const long long modn=1000;
int n;
char a[51]={};
char b[100*maxn]={};
struct trie{
    int next[26];
    bool exist;
    int count;
    int fail;
}e[maxn*50]={};
int tot,ans;
int q[maxn*50]={};

void init(int x,int k,int j){
    if(j>k){ 
        e[x].exist=1;
        e[x].count++;
        return; 
    }
    int z=a[j]-'a';
    if(!e[x].next[z])
        e[x].next[z]=++tot;
    init(e[x].next[z],k,j+1);
}
void fir(){
    memset(e,0,sizeof(e)); memset(q,0,sizeof(q));
    tot=ans=0;
}
void doit(){
    int head=0,tail=0,x,y,f;
    q[0]=0;
    while(head<=tail){
        x=q[head++];
        for(int i=0;i<26;i++){
            if(e[x].next[i]){
                y=e[x].next[i];
                if(x!=0){
                    f=e[x].fail;
                    while((!e[f].next[i]) && f )
                        f=e[f].fail;
                    e[y].fail=e[f].next[i];
                }q[++tail]=y;
            }
        }
    }
}
void find(){
    scanf("%s",&b);
    int k=std::strlen(b);
    int x=0,z,y;
    for(int i=0;i<k;i++){
        z=b[i]-'a';
        while( (!e[x].next[z])&& x){
            x=e[x].fail;
        }x=e[x].next[z]; y=x;
        while(y&&e[y].count){
            ans+=e[y].count;
            e[y].count=0;
            y=e[y].fail;
        }
    }
}
int main(){
    int T;scanf("%d",&T);
    while(T-->0){
        fir();
        scanf("%d",&n);
        for(int i=1;i<=n;i++){ 
            scanf("%s",&a);
            init(0,std::strlen(a)-1,0); 
        }
        doit();
        find();
        printf("%d
",ans);
    }
    return 0;
}

 更新：

整理模板的时候发现自己原来的代码有问题。这个新版本应该没问题了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=10100;
int n,siz;
char ch[60]={};
char str[maxn*100]={};
struct trie{
    int sig[26];
    int cnt;
    int vis;
    int fail;
}t[maxn*60];int tot=0;
int q[maxn*60]={};int head=0,tail=0;
void fir(){
    tot=0;memset(t,0,sizeof(t));//memset(q,0,sizeof(q));
}
void init(int x,int j){
    if(j>siz-1){ t[x].cnt++; return; }
    int z=ch[j]-'a';
    if(!t[x].sig[z])t[x].sig[z]=++tot;
    init(t[x].sig[z],j+1);
}
void build_fail(){
    int head=0,tail=0,x,y,f;q[0]=0;
    while(head<=tail){
        x=q[head++];
        for(int i=0;i<26;i++)
            if(t[x].sig[i]){
                y=t[x].sig[i];
                if(x){
                    f=t[x].fail;
                    while((!t[f].sig[i])&&f)
                        f=t[f].fail;
                    t[y].fail=t[f].sig[i];
                }q[++tail]=y;
            }
    }
}
int get_num(){//字符串中包含的单词数量，重复的不记录，
    //如果单词x在字符串中出现一次而在单词表中有两个则ans+2
    //在字符串中出现两次而单词表中有一个则ans+1
    int ans=0,x=0,y,z;
    for(int i=0;i<siz;i++){
        z=str[i]-'a';
        while((!t[x].sig[z])&&x)
            x=t[x].fail;
        x=t[x].sig[z];y=x;
        while(y&&(!t[y].vis)){//保证了每个结尾只访问一次
            ans+=t[y].cnt;
            t[y].vis=1;t[y].cnt=0;
            y=t[y].fail;
        }
    }
    return ans;
}
int main(){
    int T;scanf("%d",&T);
    while(T-->0){
        fir();
        scanf("%d",&n);
        for(int i=1;i<=n;i++){scanf("%s",ch);siz=strlen(ch);init(0,0);}
        build_fail();
        scanf("%s",str);siz=strlen(str);
        printf("%d
",get_num());
    }
    return 0;
}