ac自动机匹配时,建立两个栈,其中一个栈存储答案,另一个栈存储失配指针。
如果在某点找到我们要删除的单词(全部),那么我们要将匹配指针前移至此单词前面那个字母,即将我们的指针前移长度单位。
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <queue>
#include <cstring>
#include <algorithm>
#include <string>
using namespace std;
//由于题目中子串美丽度计算时不可重复,因此记录每个字符串长度,
struct Tire{
int nxt[1000100][26],fail[1000100],ed[1000100],los[1000100];
int cnt,rt;
int ans[1000100];
int newnode()
{
for(int i=0;i<26;i++)
{
nxt[cnt][i]=-1;
}
ed[cnt++]=-1;
return cnt-1;
}
void init()
{
cnt=0;
rt=newnode();
}
void ins(string s,int id)
{
int len=s.size();
int now=rt;
for(int i=0;i<len;i++)
{
if(nxt[now][s[i]-'a']==-1)
{
nxt[now][s[i]-'a']=newnode();
}
now=nxt[now][s[i]-'a'];
}
ed[now]=id;
}
void build()
{
queue<int>q;
fail[rt]=rt;
for(int i=0;i<26;i++)
{
if(nxt[rt][i]==-1)
{
nxt[rt][i]=rt;
}
else
{
fail[nxt[rt][i]]=rt;
q.push(nxt[rt][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(string s)
{
int now=rt,top=0;
for(int i=0;i<s.size();i++)
{
los[i]=now=nxt[now][s[i]-'a'];
ans[++top]=i;
if(ed[now]!=-1)
{
top-=ed[now];
now=los[ans[top]];
}
}
for(int i=1;i<=top;i++)
printf("%c",s[ans[i]]);
puts("");
}
};
Tire ac;
string s,s1;
int main()
{
int n;
cin>>s1;
scanf("%d",&n);
ac.init();
for(int i=0;i<n;i++)
{
cin>>s;
ac.ins(s,s.size());
}
ac.build();
ac.query(s1);
return 0;
}