暑假- ac自动机-(A - Keywords Search）

最新推荐文章于 2023-07-18 13:58:20 发布

slime_kirito

最新推荐文章于 2023-07-18 13:58:20 发布

阅读量313

点赞数

分类专栏：练习-训练文章标签： ac自动机

本文链接：https://blog.csdn.net/slime_kirito/article/details/47124275

版权

练习-训练专栏收录该内容

71 篇文章 0 订阅

订阅专栏

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXM=500010;
const int MAXN=1000010;
char str[MAXN];
struct trie
{
	int root,trieN;
	int child[MAXM][26],value[MAXM],fail[MAXM];
	int NewTrie()
	{
		value[trieN]=0;
		memset(child[trieN],-1,sizeof(child[trieN]));
		return trieN++;
	}
	void init()
	{
		trieN=0;
		root=NewTrie();
	}
	void Insert(char s[])
	{
		int x=root;
		for(int i=0;s[i];i++)
		{
			if(child[x][s[i]-'a']==-1)
			{
				child[x][s[i]-'a']=NewTrie();
			}
			x=child[x][s[i]-'a'];
		}
		value[x]++;
	}
	void Build()
	{
		int x;
		queue<int> q;
		fail[root]=root;
		for(int i=0;i<26;i++)
		{
			if(child[root][i]==-1)
			{
				child[root][i]=root;
			}
			else
			{
				fail[child[root][i]]=root;
				q.push(child[root][i]);
			}
		}
		while(!q.empty())
		{
			x=q.front();q.pop();
			for(int i=0;i<26;i++)
			{
				if(child[x][i]==-1)
				{
					child[x][i]=child[fail[x]][i];
				}
				else
				{
					fail[child[x][i]]=child[fail[x]][i];
					q.push(child[x][i]);
				}
			}
		}
	}	
	int query(char s[])
	{
		int sum=0;
		int x=root,temp;
		for(int i=0;s[i];i++)
		{
			temp=x=child[x][s[i]-'a'];
			while(temp!=root)
			{
				sum+=value[temp];
				value[temp]=0;
				temp=fail[temp];
			}
		}
		return sum;
	}
}ac;
int main()
{
	int t,n;
	cin>>t;
	while(t--)
	{
		cin>>n;
		char ch[55];
		ac.init();
		for(int i=0;i<n;i++)
		{
			cin>>ch;
			ac.Insert(ch);
		}
		ac.Build();
		cin>>str;
		cout<<ac.query(str)<<endl;
	}
	return 0;
}