hdu 2222 Keywords Search

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

第一个AC自动机

AC自动机就是在trie上进行kmp

需要三个步骤

1，建立trie

2，求fail指针

3，匹配

代码：

#include<iostream>

#include<cmath>

#include<cstdio>

#include<string>

#include<cstring>

#include<vector>

#include<stack>

#include<queue>

#include<set>

#include<map>

#include<algorithm>



#define LL long long



using namespace std;



const int INF=0x3f3f3f3f;

const int N=1000005;

const int K=26;

struct nodeTrie

{

    nodeTrie()

    {

        v=0;

        fail=NULL;

        for(int i=0;i<K;++i)

        next[i]=NULL;

    }

    int v;

    struct nodeTrie *fail;

    struct nodeTrie *next[K];

}*root;

char s[N];

void addWord(nodeTrie *p,char *s)

{

    if(s[0]=='\0') return ;

    for(int i=0;s[i]!='\0';++i)

    {

        if(p->next[s[i]-'a']==NULL)

        p->next[s[i]-'a']=new nodeTrie;

        p=p->next[s[i]-'a'];

    }

    ++(p->v);

}

void init(int n)

{

    root=new nodeTrie;

    while(n--)

    {

        gets(s);

        addWord(root,s);

    }

}

void bfs(nodeTrie *p)

{

    p->fail=root;

    queue<nodeTrie *>qt;

    qt.push(p);

    while(!qt.empty())

    {

        nodeTrie *y;

        nodeTrie *x=qt.front();qt.pop();

        for(int i=0;i<K;++i)

        if(x->next[i]!=NULL)

        {

            qt.push(x->next[i]);

            if(x==root)

            {x->next[i]->fail=root;continue;}

            y=x->fail;

            while(y!=root&&y->next[i]==NULL)

            y=y->fail;

            if(y->next[i]!=NULL)

            x->next[i]->fail=y->next[i];

            else

            x->next[i]->fail=root;

        }

    }

}

int match(nodeTrie *p,char *s)

{

    int wordCount=0;

    int l=0;

    while(s[l]!='\0')

    {

        while(p->next[s[l]-'a']==NULL&&p!=root)

        p=p->fail;

        if(p->next[s[l]-'a']!=NULL)

        p=p->next[s[l]-'a'];

        ++l;

        nodeTrie *fp=p;

        while(fp!=root)

        {

            if((fp->v)>0) {wordCount+=(fp->v);fp->v=0;}

            fp=fp->fail;

        }

    }

    return wordCount;

}

int main()

{

    //freopen("data.in","r",stdin);

    int T;

    scanf("%d",&T);

    while(T--)

    {

        int n;

        scanf("%d ",&n);

        init(n);

        bfs(root);

        gets(s);

        printf("%d\n",match(root,s));

    }

    return 0;

}