HDU2222 Keywords Search

考查：多串匹配 AC自动机

提交：3次TLE 一次WA 203MS

第一道AC自动机题目，AC自动机(Aho-Corasick automation)是用来解决多串匹配问题的算法。而KMP的解决单串匹配的，即在给定的文本中找到搜索的关键字和它们的位置，而如果需要在一段给定的文本中同时搜索多个关键字的话就需要用到AC自动机了。

AC自动机是基于Trie实现的，首先需要建立一颗关键字的Trie树，这里需要注意的是为要搜索的关键字建树而不是文本串，在建立完树后就需要建立fail指针。假设有一个节点k，他的失败指针指向j，那么k,j满足这个性质：设root到j的距离为n，则从k之上的第n个节点到k所组成的长度为n的单词，与从root到j所组成的单词相同。还有很多问题不是很明白，不过这里还有很多很好的介绍AC自动机的文章，大家可以看看：

http://www.cppblog.com/mythit/archive/2009/04/21/80633.html

http://www.dev-club.net/xiangxixinxi/22010072906013413/201102100439240.html

http://www.cublog.cn/u3/113538/showart_2212716.html

此外还有需要注意的一点是在循环中，不要用i<strlen(text)，之前TLE就是这个问题，应该是这样每次都会调用strlen函数的原因，可以直接写为text[i]。

#include <stdio.h> #include <stdlib.h> #include <string.h> typedef struct _n { struct _n *fail; struct _n *next[26]; int cnt; }node; char text[1000001]; node *root; node *queue[500000]; void insert(char *key) { int i,j; node *p=root; for(i=0;key[i];i++) { if(!p->next[key[i]-'a']) { p->next[key[i]-'a']=(node*)malloc(sizeof(node)); memset(p->next[key[i]-'a'],0,sizeof(node)); } p=p->next[key[i]-'a']; } p->cnt++; } void ac_auto() { int i,front,back; node *tmp,*p; front=back=0; queue[back++]=root; while(front!=back) { tmp=queue[front++]; for(i=0;i<26;i++) { if(tmp->next[i]) { if(tmp==root) tmp->next[i]->fail=root; else { p=tmp->fail; while(p!=NULL) { if(p->next[i]!=NULL) { tmp->next[i]->fail=p->next[i]; break; } p=p->fail; } if(p==NULL) tmp->next[i]->fail=root; } queue[back++]=tmp->next[i]; } } } } int match() { int i,idx,r; node *tmp; node *p=root; r=0; for(i=0;text[i];i++) { idx=text[i]-'a'; while(p->next[idx]==NULL&&p!=root) p=p->fail; p=p->next[idx]; p=(p==NULL)?root:p; tmp=p; while(tmp!=root&&tmp->cnt!=-1) { r+=tmp->cnt; tmp->cnt=-1; tmp=tmp->fail; } } return r; } int main() { int c,n,i; char key[51]; scanf("%d",&c); while(c--) { root=(node*)malloc(sizeof(node)); root->fail=NULL; root->cnt=0; for(i=0;i<26;i++) root->next[i]=NULL; scanf("%d",&n); for(i=0;i<n;i++) { scanf("%s",key); insert(key); } ac_auto(); scanf("%s",text); printf("%d/n",match()); } return 0; }