《ybtoj高效进阶》第二部分第五章例题1 单词查询

题目大意

AC自动机板子,多组数据

思路

不说了,详见代码
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue> 
using namespace std;
int trie[1000006][26],tot=1,ed[1000006],nxt[1000006],n,m;
string x;
void insert(string x)
{
    
    
	int p=1;
	for (int i=0;i<x.size();i++)
	{
    
    
		if (!trie[p][x[i]-'a']) trie[p][x[i]-'a']=++tot,memset(trie[tot],0,sizeof(trie[tot]));
		p=trie[p][x[i]-'a'];
	}
	ed[p]++;
	return;
}
void bfs()
{
    
    
	queue<int> o;
	for (int i=0;i<26;i++) trie[0][i]=1;
	o.push(1);
	nxt[1]=0;
	while (o.size())
	{
    
    
		int u=o.front();
		o.pop();
		for (int i=0;i<26;i++)
		{
    
    
			if (!trie[u][i]) trie[u][i]=trie[nxt[u]][i];
			else
			{
    
    
				nxt[trie[u][i]]=trie[nxt[u]][i];
				o.push(trie[u][i]);
			}
		}
	}
	return;
}
int ACQuantAsk(string x)
{
    
    
	int ans=0,u=1,l=x.size();
	for (int i=0;i<l;i++)
	{
    
    
		int k=trie[u][x[i]-'a'];
		while (k>1&&ed[k]!=-1)
		{
    
    
			ans+=ed[k];
			ed[k]=-1;
			k=nxt[k];
		}
		u=trie[u][x[i]-'a'];
	}
	return ans;
}
int main()
{
    
    
	int t;
	cin>>t;
	while (t--)
	{
    
    
		memset(ed,0,sizeof(ed));
		tot=1;
		memset(nxt,0,sizeof(nxt));
		for (int i=0;i<26;i++) trie[0][i]=1,trie[1][i]=0;
		cin>>n;
		for (int i=1;i<=n;i++)
		{
    
    
			cin>>x;
			insert(x);
		}
		bfs();
		cin>>x;
		cout<<ACQuantAsk(x)<<endl;
	}
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_49843717/article/details/115254114
今日推荐