hdu 2222 ac自动机模板题

最新推荐文章于 2024-04-26 20:01:35 发布

圣诞老人家

最新推荐文章于 2024-04-26 20:01:35 发布

阅读量545

点赞数

CC 4.0 BY-SA版权

分类专栏： ac自动机

本文链接：https://blog.youkuaiyun.com/tianyuhang123/article/details/56679113

ac自动机专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种使用AC自动机进行高效关键词匹配的方法，应用于图像描述的检索系统中。通过构建Trie树并完善AC自动机，实现了快速查找关键词在长文本中的出现次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

In the modern time, Search engine came into the life of everybody like Google, Baidu, etc.
Wiskey also wants to bring this feature to his image retrieval system.
Every image have a long description, when users type some keywords to find the image, the system will match the keywords with description of image and show the image which the most keywords be matched.
To simplify the problem, giving you a description of image, and some keywords, you should tell me how many keywords will be match.

Input

First line will contain one integer means how many cases will follow by.
Each case will contain two integers N means the number of keywords and N keywords follow. (N <= 10000)
Each keyword will only contains characters 'a'-'z', and the length will be not longer than 50.
The last line is the description, and the length will be not longer than 1000000.

Output

Print how many keywords are contained in the description.

Sample Input

1
5
she
he
say
shr
her
yasherhs

Sample Output

题意：给你多个串，然后给你一个长串，问长串中含有几个所给短串

思路：ac自动机模板题如果不了解该算法可以先看一下这个http://blog.youkuaiyun.com/tianyuhang123/article/details/56679075

ac代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct Node
{
int cnt;//是否为该单词的最后一个结点
Node *fail;//失败指针
Node *next[26];//Trie中每个结点的各个节点
}*queue[500005];//队列，方便用BFS构造失败指针
char s[1000005];//主字符串
char keyword[55];//需要查找的单词
Node *root;//头结点
void Init(Node *root)//每个结点的初始化
{
root->cnt=0;
root->fail=NULL;
for(int i=0;i<26;i++)
root->next[i]=NULL;
}
void Build_trie(char *keyword)//构建Trie树
{
Node *p,*q;
int i,v;
int len=strlen(keyword);
for(i=0,p=root;i<len;i++)
{
v=keyword[i]-'a';
if(p->next[v]==NULL)
{
q=(struct Node *)malloc(sizeof(Node));
Init(q);
p->next[v]=q;//结点链接
}
p=p->next[v];//指针移动到下一个结点
}
p->cnt++;//单词最后一个结点cnt++，代表一个单词
}
void Build_AC_automation(Node *root)
{
int head=0,tail=0;//队列头、尾指针
queue[head++]=root;//先将root入队
while(head!=tail)
{
Node *p=NULL;
Node *temp=queue[tail++];//弹出队头结点
for(int i=0;i<26;i++)
{
if(temp->next[i]!=NULL)//找到实际存在的字符结点
{ //temp->next[i] 为该结点，temp为其父结点
if(temp==root)//若是第一层中的字符结点，则把该结点的失败指针指向root
temp->next[i]->fail=root;
else
{
//依次回溯该节点的父节点的失败指针直到某节点的next[i]与该节点相同，
//则把该节点的失败指针指向该next[i]节点；
//若回溯到 root 都没有找到，则该节点的失败指针指向 root
p=temp->fail;//将该结点的父结点的失败指针给p
while(p!=NULL)
{
if(p->next[i]!=NULL)
{
temp->next[i]->fail=p->next[i];
break;
}
p=p->fail;
}
//让该结点的失败指针也指向root
if(p==NULL)
temp->next[i]->fail=root;
}
queue[head++]=temp->next[i];//每处理一个结点，都让该结点的所有孩子依次入队
}
}
}
}
int query(Node *root)
{ //i为主串指针，p为模式串指针
int i,v,count=0;
Node *p=root;
int len=strlen(s);
for(i=0;i<len;i++)
{
v=s[i]-'a';
//由失败指针回溯查找，判断s[i]是否存在于Trie树中
while(p->next[v]==NULL && p!=root)
p=p->fail;
p=p->next[v];//找到后p指针指向该结点
if(p==NULL)//若指针返回为空，则没有找到与之匹配的字符
p=root;
Node *temp=p;//匹配该结点后，沿其失败指针回溯，判断其它结点是否匹配
while(temp!=root)//匹配结束控制
{
if(temp->cnt>=0)//判断该结点是否被访问
{
count+=temp->cnt;//由于cnt初始化为 0，所以只有cnt>0时才统计了单词的个数
temp->cnt=-1;//标记已访问过
}
else//结点已访问，退出循环
break;
temp=temp->fail;//回溯失败指针继续寻找下一个满足条件的结点
}
}
return count;
}
int main()
{
int T,n;
scanf("%d",&T);
while(T--)
{
root=(struct Node *)malloc(sizeof(Node));
Init(root);
scanf("%d",&n);
getchar();
for(int i=0;i<n;i++)
{
gets(keyword);
Build_trie(keyword);
}
Build_AC_automation(root);
gets(s);
printf("%d\n",query(root));
}
return 0;
}