【USACO 15 FEB】审查

传送门

---

Problem

给出 $n$ 个单词 $S_1,S_2,...,S_n$ 和一个长度不超过 $10^5$ 的字符串 $T$，你需要从 $T$ 中删掉这 $n$ 个单词。

现在每次在 $T$ 中找到最早出现的列表中的单词（最早出现指该单词的开始位置最小），然后从 $T$ 中删除这个单词。他重复这个操作直到 $T$ 中没有列表里的单词为止。

保证列表中的单词不会出现一个单词是另一个单词子串的情况，这意味着每个列表中的单词在 $T$ 中出现的开始位置是互不相同的。

注意删除一个单词后可能会导致 $T$ 中出现另一个列表中的单词。

请完成这些操作并输出最后的 $T$。

---

Solution

还是蛮简单的一道题。

先把 $n$ 个单词的AC自动机建出来，考虑怎么删单词。

我们用一个栈来维护匹配到某一位置时 $T$ 中还剩下的单词（就是没被删掉的）。

然后就用 $T$ 在AC自动机上匹配，依次在栈中添加字母，每当匹配到一个单词表上的单词时，就在栈中把这个单词删掉（就直接把 $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;
}