hdu 2222 AC自动机

最新推荐文章于 2020-09-14 18:22:41 发布

原创最新推荐文章于 2020-09-14 18:22:41 发布 · 261 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#hdu #AC自动机

本文介绍了一种使用AC自动机处理字符串匹配问题的方法。通过构造AC自动机，可以高效地完成多个模式串在文本中的查找。文章详细展示了AC自动机的数据结构定义、初始化、插入模式串、计算失败指针及查询过程，并提供了完整的C++实现代码。

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

#include<cstdio>
#include<cstring>
#include<queue>
#define N 26
#define MAX_N 600000
using namespace std;
struct tire
{
	int next[MAX_N][N],fail[MAX_N],sum[MAX_N];
	int root,tot;
	int newnode()
	{
		for(int i=0;i<N;i++)
		next[tot][i]=-1;
		sum[tot++]=0;
		return tot-1;
	} 
	void init()
	{
		tot=0;
		root=newnode();
	}
	void insert(char *p)
	{
		int len=strlen(p);
		int now=root;
		for(int i=0;i<len;i++)
		{
			int tem=p[i]-'a';
			if(next[now][tem]==-1)
			next[now][tem]=newnode();
			now=next[now][tem];
		}
		sum[now]++;
	}
	void cal()
	{
		queue<int>q;
		fail[root]=root;
		for(int i=0;i<N;i++)
		if(next[root][i]==-1)
		next[root][i]=root;
		else
		{
			fail[next[root][i]]=root;
			q.push(next[root][i]);
		}
		while(!q.empty())
		{
			int now=q.front();
			q.pop();
			for(int i=0;i<N;i++)
			{
				if(next[now][i]==-1)
				next[now][i]=next[fail[now]][i];
				else
				{
					fail[next[now][i]]=next[fail[now]][i];
					q.push(next[now][i]);
				}
			}
		}
	}
		int query(char *s)
		{
			int len=strlen(s);
			int now=root;
			int res=0;
			for(int i=0;i<len;i++)
			{
				now=next[now][s[i]-'a'];
				int temp=now;
				while(temp!=root)
				{
					res+=sum[temp];
					sum[temp]=0;
					temp=fail[temp];
				}
			}
			return res;
		}
};
tire d;
char str[1000010];
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int n;
		scanf("%d",&n);
		d.init();
		for(int i=0;i<n;i++)
		{
			scanf("%s",str);
			d.insert(str);
		}
		d.cal();
		scanf("%s",str);
		printf("%d\n",d.query(str));
	}
}