《ybtoj高效进阶》第二部分第五章例题3 屏蔽词删除

题目大意

有一个字符串,然后又一些指定的小串。
你会从头开始找到第一个字符串的子串使得它是某个小串,然后把这个串从字符串中删除。
然后继续重复这个操作,知道找不到这些小串在字符串中出现。
问你最后的字符串。

思路

把屏蔽词插入自动机,然后跑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[100005];
char stik[1000006];
int stck[1000006];
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]=x.size();
	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;
}
void ACQuantAsk(string x)
{
    
    
	int ans=0,u=1,l=x.size();
	for (int i=0;i<l;i++)
	{
    
    
		stik[++ans]=x[i];
		u=trie[u][x[i]-'a'];
		stck[ans]=u;
		if (ed[u])
		{
    
    
			ans-=ed[u];
			u=stck[ans];
		}
	}
	for (int i=1;i<=ans;i++) cout<<stik[i];
	return;
}
string xxx;
int main()
{
    
    
	for (int i=0;i<26;i++) trie[0][i]=1,trie[1][i]=0;
	cin>>xxx;
	cin>>n;
	for (int i=1;i<=n;i++)
	{
    
    
		cin>>x[i];
		insert(x[i]);
	}
	bfs();
	ACQuantAsk(xxx);
    return 0;
}

猜你喜欢

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