hdu2222Keywords Search(AC自动机模板题)

题意：

给你n个模板串，一个文本串，求有几个模板串出现在文本串中。

思路：

如果暴力n次kmp，稳超时。所以把模板串处理一下，设置fail指针（相当与kmp里的next)。ac自动机可以处理多模式匹配，要注意的是，多模式匹配时，匹配到一个串时还可能同时匹配到别的串。

代码：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;
const int maxn = 1000000+7;
char key[57];
struct node
{
int val;
node *next[26];
node *fail;
};

void ins(node *root,char *p)
{
int len = strlen(p);
int i = 0,id;
while(i<len)
{
id = p[i]-'a';
if(root->next[id]==NULL)
{
node *temp = new node;
temp->fail = NULL;
temp->val = 0;
for(int i = 0;i<26;i++)
temp->next[i] = NULL;
root->next[id] = temp;
}
root = root->next[id];
i++;
}
root->val++;
}

void build(node *root)
{
queue<node*> q;
q.push(root);
while(!q.empty())
{
node *temp = q.front();
q.pop();
for(int i = 0;i<26;i++)
{
if(temp->next[i]!=NULL)
{
if(temp==root)
{
temp->next[i]->fail = root;
}
else
{
node *k = temp->fail;
while(k)
{
if(k->next[i]!=NULL)
{
temp->next[i]->fail = k->next[i];
break;
}
k = k->fail;
//cout<<"hh"<<endl;
}
if(!k)
temp->next[i]->fail = root;
}
q.push(temp->next[i]);
}
}
//cout<<q.size()<<endl;
}
}

int add(node *temp) //匹配到当前串的同时，可能还同时会匹配到别的串
{
int sum = 0;
while(temp!=NULL&&temp->val>=0)
{
sum += temp->val;
temp->val = -1;
temp = temp->fail;
}
return sum;
}

int match(node *root,char *str)
{
int cnt = 0;
int len = strlen(str);
int id;
node *p = root;
for(int i = 0;i<len;i++)
{
id = str[i]-'a';
while(p!=root&&p->next[id]==NULL)
p = p->fail;
if(p->next[id]!=NULL)
{
p = p->next[id];
cnt += add(p);
}
}
return cnt;
}

void del(node *root)
{
for(int i = 0;i<26;i++)
if(root->next[i]!=NULL)
del(root->next[i]);
delete root;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
getchar();
node *root = new node;
root->val = 0;
root->fail = NULL;
for(int i = 0;i<26;i++)
root->next[i] = NULL;
for(int i = 0;i<n;i++)
{
scanf("%s",key);
ins(root,key);
}
build(root);
//cout<<"kii"<<endl;
char str[maxn];
scanf("%s",str);
printf("%d\n",match(root,str));
del(root);
}
return 0;
}