HDU 2222 Keywords Search （AC 自动机入门）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2222

这个题目很久以前写过，不过好久没接触AC自动机的题了，又拿过来熟悉一下！

题目有一点要注意，也就是给的串可能会重复！

这次写的代码貌似比上一次写的简洁多了～

#include <string.h>
#include <algorithm>
#include <stdio.h>
#include <iostream>
#include <queue>
using namespace std;
#define maxn 1000010
char str[maxn],s[100];
int n,pos;
struct trie{
    trie *next[26];
    int num;
    trie *fail;
}po[1000000],re_root;
queue<trie *> q;
int insert_trie(trie *root,char *name){
    if(name[0]==0){
        root->num++;
        return 0;
    }
    int t=name[0]-'a';
    if(root->next[t]==NULL) { memset(po+pos,0,sizeof(trie));root->next[t]=&po[pos++];}
    insert_trie(root->next[t],name+1);
    return 0;
}
int build_ac_auto(){
    q.push(&re_root);
    re_root.fail=NULL;
    trie *p,*temp;
    int i,j,k;
    while(!q.empty()){
        p=q.front(),q.pop();
        for(i=0;i<26;i++){
            if(p->next[i]){
                temp=p->fail;
                while(temp && temp->next[i]==NULL) temp=temp->fail;
                if(temp==NULL) p->next[i]->fail=&re_root;
                else p->next[i]->fail=temp->next[i];
                q.push(p->next[i]);
            }
        }
    }
    return 0;
}
int query(trie *root,char *name){
    int i=0,ans=0,t;
    trie *p=root,*temp;
    while(name[i]){
        t=name[i]-'a';
        while(p!=NULL && p->next[t]==NULL) p=p->fail;
        if(p==NULL) p=&re_root;
        else p=p->next[t];
        temp=p;
        while(temp && temp->num!=-1){
            ans+=temp->num;
            temp->num=-1;
            temp=temp->fail;
        }
        i++;
    }
    printf("%d\n",ans);
    return 0;
}
int main(){
    int i,j,k,t;
    scanf("%d",&t);
    while(t--){
        pos=0;
        memset(&re_root,0,sizeof(re_root));
        scanf("%d",&n);
        for(i=0;i<n;i++){
            scanf("%s",s);
            insert_trie(&re_root,s);
        }
        build_ac_auto();
        scanf("%s",str);
        query(&re_root,str);
    }
    return 0;
}