Ac自动机_hdu_2222_Keywords Search

题目连接

这是一道AC自动机模板题，关于Ac自动机我总结一小点，因为对于字典树，它的难点就在于构建fail指针，要是fail指针会构建了Ac自动机也就基本上叫懂了。而相对于Kmp算法，AC自动机构建fail指针又有点不同，这里是站在父节点上构建子节点的fail指针，因为每次匹配都是站在父节点上找它是否存在某个子节点，因此当某个时刻某个节点不存在要找的下一个子节点时，就跳到它的fail指针处看那里是否可以匹配，因此当前节点fail指针指向的节点所表示的字符一定与当前节点相同，至于哪个是哪个前缀后缀什么的会了kmp的话就好理解了。。

语文好菜。。讲不下去了，上几个我学的时候用的链接吧，自己看看还是很好懂得。

KMP&AC自动机

AC自动机算法

代码君

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <string>
#include <cstdio>
using namespace std;

struct node
{
int coun;
node *fail;
node *next[26];
node(){
coun=0;
fail=NULL;
memset(next,0,sizeof(next));
}
};

class Trie
{
private:
node *root;
public:
Trie(){
root=new node();
}
~Trie(){
del(root);
}
void del(node *p){
if(p==NULL)return;
for(int i=0;i<26;i++)del(p->next[i]);
delete p;
}
void insert(string s){
node *p=root;
int len=s.length();
for(int i=0;i<len;i++){
int id=s[i]-'a';
if(p->next[id]==NULL){
p->next[id]=new node();
}
p=p->next[id];
}
p->coun++;
}
void build(){
queue<node*>Q;
Q.push(root);
while(!Q.empty()){
node *p=Q.front();
Q.pop();
for(int i=0;i<26;i++){
if(p->next[i]!=NULL){
if(p==root)p->next[i]->fail=root;
else{
node *tmp=p->fail;
while(tmp!=root&&tmp->next[i]==NULL)tmp=tmp->fail;
tmp=tmp->next[i];
if(tmp==NULL)tmp=root;
p->next[i]->fail=tmp;
}
Q.push(p->next[i]);
}
}
}
}
int query(string s){
int ans=0;
node *p=root;
int len=s.length();
for(int i=0;i<len;i++){
int id=s[i]-'a';
while(p!=root && p->next[id]==NULL)p=p->fail;
p=p->next[id];
if(p==NULL)p=root;
node *tmp=p;
while(tmp!=root){
if(tmp->coun>=0){
ans+=tmp->coun;
tmp->coun=-1;
}else break;
tmp=tmp->fail;
}
}
return ans;
}
};

int main(){
int T;
cin>>T;
while(T--){
int n;
string s;
scanf("%d",&n);
Trie *ac=new Trie();
for(int i=0;i<n;i++){
cin>>s;
ac->insert(s);
}
cin>>s;
ac->build();
printf("%d\n",ac->query(s));
}
return 0;
}