hdu 2222（初级AC自动机）

最新推荐文章于 2019-08-22 09:44:00 发布

slege

最新推荐文章于 2019-08-22 09:44:00 发布

阅读量669

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/yangyafeiac/article/details/9319611

本文详细介绍了模版级别的AC自动机的概念、实现过程及应用。通过代码示例展示了如何构建自动机并进行字符串匹配操作。

点击打开链接

模版级别的AC自动机！！！！

#include"stdio.h"
#include"string.h"
#define N 500004
const int kind=26;
struct node
{
	node *fail;
	node *next[kind];
	int count;
	node()
	{
		fail=NULL;
		count=0;
		memset(next,NULL,sizeof(next));
	}
}*q[N];
char keyword[51];
char str[N*2];
int head,tail;
void insert(char *str,node *root)
{
	node *p=root;
	int i=0,index;
	while(str[i])
	{
		index=str[i]-'a';
		if(p->next[index]==NULL)p->next[index]=new node();
		p=p->next[index];
		i++;
	}
	p->count++;
}
void build_ac_automation(node *root)
{
	int i;
	root->fail=NULL;
	q[head++]=root;
	while(head!=tail)
	{
		node *temp=q[tail++];
		node *p=NULL;
		for(i=0;i<26;i++)
		{
			if(temp->next[i]!=NULL)
			{
				if(temp==root)temp->next[i]->fail=root;
				else
				{
					p=temp->fail;
					while(p!=NULL)
					{
						if(p->next[i]!=NULL)
						{
							temp->next[i]->fail=p->next[i];
							break;
						}
						p=p->fail;
					}
					if(p==NULL)temp->next[i]->fail=root;
				}
				q[head++]=temp->next[i];
			}
		}
	}
}

int query(node *root)
{
	int i=0,cnt=0,index,len=strlen(str);
	node *p=root;
	while(str[i])
	{
		index=str[i]-'a';
		while(p->next[index]==NULL&&p!=root)p=p->fail;
		p=p->next[index];
		p=(p==NULL)?root:p;
		node *temp=p;
		while(temp!=root&&temp->count!=-1)
		{
			cnt+=temp->count;
			temp->count=-1;
			temp=temp->fail;
		}
		i++;
	}
	return cnt;
}
int main()
{
	int n,t;
	scanf("%d",&t);
	while(t--)
	{
		head=tail=0;
		node *root=new node();
		scanf("%d",&n);
		getchar();
		while(n--)
		{
			gets(keyword);
			insert(keyword,root);
		}
		build_ac_automation(root);
		scanf("%s",str);
		printf("%d\n",query(root));
	}
	return 0;
}