题目链接:Here!
题目大意:给你几个字符串再给你一段字符,让你把字符中所有出现的字符串都变成*,最后输出处理后的字符串
题目思路:AC自动机直接处理出每个字符串的位置,直接替换即可。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <string>
#include <sstream>
#include <vector>
#include <map>
#include <queue>
#include <stack>
#include <set>
#include <list>
#include <deque>
#define LL long long
#define INF 0x3f3f3f3f
#define PII pair<int,int>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define lowbit(x) (x&(-x))
#define mem(a,b) memset((a),(b),sizeof(a))
using namespace std;
vector<int>pos;
struct Trie{
int nxt[1000010][26],fail[1000010];
pair<int,int>ed[1000010];
int root,L;
int newnode()
{
for(int i=0;i<26;i++)
nxt[L][i]=-1;
ed[L].fi=ed[L++].se=0;
return L-1;
}
void init()
{
L=0;
root=newnode();
}
void Insert(char buf[])
{
int len=strlen(buf);
int now=root;
for(int i=0;i<len;i++)
{
if(nxt[now][buf[i]-'a']==-1)
nxt[now][buf[i]-'a']=newnode();
now=nxt[now][buf[i]-'a'];
}
ed[now].fi++;
ed[now].se=len;
}
void build()
{
queue<int>Q;
fail[root]=root;
for(int i=0;i<26;i++)
if(nxt[root][i]==-1)
nxt[root][i]=root;
else
{
fail[nxt[root][i]]=root;
Q.push(nxt[root][i]);
}
while(!Q.empty())
{
int now=Q.front();
Q.pop();
for(int i=0;i<26;i++)
if(nxt[now][i]==-1)
nxt[now][i]=nxt[fail[now]][i];
else
{
fail[nxt[now][i]]=nxt[fail[now]][i];
Q.push(nxt[now][i]);
}
}
}
void query(char buf[])
{
int len=strlen(buf);
int now=root;
int res=0;
for(int i=0;i<len;i++)
{
now=nxt[now][buf[i]-'a'];
int temp=now,len=0;
while(temp!=root)
{
res+=ed[temp].fi;
ed[temp].fi=0;
len=max(len,ed[temp].se);
temp=fail[temp];
}
for(int j=i;j>=i-len+1;j--)
buf[j]='*';
}
for(int i=0;i<pos.size();i++)
if(buf[pos[i]]!='*')
buf[pos[i]]-=32;
printf("%s\n",buf);
}
}ac;
char tmp[1000005];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
ac.init();
pos.clear();
getchar();
for(int i=1;i<=n;i++)
{
gets(tmp);
ac.Insert(tmp);
}
ac.build();
gets(tmp);
int len=strlen(tmp);
for(int i=0;i<len;i++)
if(tmp[i]>='A'&&tmp[i]<='Z')
{
tmp[i]+=32;
pos.pb(i);
}
ac.query(tmp);
}
return 0;
}