Problem
给出 n n n 个单词 S 1 , S 2 , . . . , S n S_1,S_2,...,S_n S1,S2,...,Sn 和一个长度不超过 1 0 5 10^5 105 的字符串 T T T,你需要从 T T T 中删掉这 n n n 个单词。
现在每次在 T T T 中找到最早出现的列表中的单词(最早出现指该单词的开始位置最小),然后从 T T T 中删除这个单词。他重复这个操作直到 T T T 中没有列表里的单词为止。
保证列表中的单词不会出现一个单词是另一个单词子串的情况,这意味着每个列表中的单词在 T T T 中出现的开始位置是互不相同的。
注意删除一个单词后可能会导致 T T T 中出现另一个列表中的单词。
请完成这些操作并输出最后的 T T T。
Solution
还是蛮简单的一道题。
先把 n n n 个单词的AC自动机建出来,考虑怎么删单词。
我们用一个栈来维护匹配到某一位置时 T T T 中还剩下的单词(就是没被删掉的)。
然后就用 T T T 在AC自动机上匹配,依次在栈中添加字母,每当匹配到一个单词表上的单词时,就在栈中把这个单词删掉(就直接把 t o p top top 减掉这个单词的长度即可),然后继续直到匹配结束。
最后直接把栈中的元素输出出来就可以了。
注意:为了方便,可以把单词的尾结点赋为这个单词的长度。
Code
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
int tot=0;
char S[N],T[N];
struct Trie{
int son[26];
int fail,end;
}a[N];
void insert(){
int i,p=0;
int l=strlen(S);
for(i=0;i<l;++i){
int x=S[i]-'a';
if(!a[p].son[x])
a[p].son[x]=++tot;
p=a[p].son[x];
}
a[p].end=l;
}
void Get_fail(){
int x,i;
queue<int>Q;
for(i=0;i<26;++i){
if(a[0].son[i]){
a[a[0].son[i]].fail=0;
Q.push(a[0].son[i]);
}
}
while(!Q.empty()){
x=Q.front();Q.pop();
for(i=0;i<26;++i){
if(a[x].son[i]){
a[a[x].son[i]].fail=a[a[x].fail].son[i];
Q.push(a[x].son[i]);
}
else a[x].son[i]=a[a[x].fail].son[i];
}
}
}
int top=0,stk[N],pos[N];
void solve(){
int i,p=0;
int l=strlen(T);
for(i=0;i<l;++i){
p=a[p].son[T[i]-'a'];
pos[++top]=p,stk[top]=i;
if(a[p].end){
top-=a[p].end;
p=top?pos[top]:0;
}
}
}
int main(){
int n,i;
scanf("%s%d",T,&n);
for(i=1;i<=n;++i)
scanf("%s",S),insert();
Get_fail(),solve();
for(i=1;i<=top;++i)
printf("%c",T[stk[i]]);
puts("");
return 0;
}