HDU 2222 Keywords Search

ac自动机模板题

链表写法：

#include<iostream>
#include<stdio.h>
#include <string.h>
using namespace std;

#define T_size 1000000
#define P_size 50
#define Total_p 10000

struct trie
{
    trie *next[26];
    trie *fail;
    int num;
    trie()
    {
        for(int i=0; i<26; i++)
        {
            next[i]=NULL;
        }
        fail=NULL;
        num=0;
    }
};

char T[T_size+1];
char P[P_size+1];
trie* q[Total_p*P_size];

void insert(trie *root,char *s)
{
    trie*p=root;
    for(int i=0; s[i]!=''; i++)
    {
        if(p->next[s[i]-'a']==NULL)
            p->next[s[i]-'a']=new trie;
        p=p->next[s[i]-'a'];
    }
    p->num++;
}

void build_ac(trie* root)
{
    int head=0,tail=0;
    q[tail++]=root;
    while(head!=tail)
    {
        trie* front=q[head++];
        for(int i=0; i<26; i++)
        {
            if(front->next[i]!=NULL)
            {
                trie*p=front->fail;
                while(p!=NULL)
                {
                    if(p->next[i]!=NULL)
                    {
                        front->next[i]->fail=p->next[i];
                        break;
                    }
                    p=p->fail;
                }
                if(p==NULL)
                    front->next[i]->fail=root;
                q[tail++]=front->next[i];
            }
        }
    }
}

int acfind(trie* root,char* T)
{
    trie*p=root;
    int sum=0;
    for(int i=0,len=strlen(T); i<len; i++)
    {
        while(p->next[T[i]-'a']==NULL&&p!=root)
        {
            p=p->fail;
        }
        if(p->next[T[i]-'a']!=NULL)
           p=p->next[T[i]-'a'];
        trie*temp=p;
        while(temp!=root&&temp->num!=-1)
        {
            sum+=temp->num;
            temp->num=-1;
            temp=temp->fail;
        }
    }
    return sum;
}

int main()
{
    int t;
    for(scanf("%d",&t); t>0; t--)
    {
        trie*root=new trie;
        int n;
        scanf("%d",&n);
        getchar();
        for(int i=0; i<n; i++)
        {
            gets(P);
            insert(root,P);
        }
        build_ac(root);
        gets(T);
        printf("%d
",acfind(root,T));
    }
    return 0;
}

数组写法：