BZOJ_3940_[Usaco2015 Feb]Censoring_AC自动机

BZOJ_3940_[Usaco2015 Feb]Censoring_AC自动机

Description

FJ把杂志上所有的文章摘抄了下来并把它变成了一个长度不超过10^5的字符串S。他有一个包含n个单词的列表，列表里的n个单词

记为t_1...t_N。他希望从S中删除这些单词。 

FJ每次在S中找到最早出现的列表中的单词(最早出现指该单词的开始位置最小)，然后从S中删除这个单词。他重复这个操作直到S中

没有列表里的单词为止。注意删除一个单词后可能会导致S中出现另一个列表中的单词 

FJ注意到列表中的单词不会出现一个单词是另一个单词子串的情况，这意味着每个列表中的单词在S中出现的开始位置是互不相同的 

请帮助FJ完成这些操作并输出最后的S

Input

第一行包含一个字符串S 

第二行包含一个整数N 

接下来的N行，每行包含一个字符串，第i行的字符串是t_i

Output

The string S after all deletions are complete. It is guaranteed that S will not become empty during the deletion process.

一行，输出操作后的S

 

Sample Input

begintheescapexecutionatthebreakofdawn
2
escape
execution

Sample Output

beginthatthebreakofdawn

---

首先是要对单词建立AC自动机。

然后用一个栈记录答案和匹配到哪个点。

因为题里说不会有一个单词会是另一个单词的子串，于是遇到被标记的结点就弹栈即可。

 

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 200050
int ch[N][26],cnt=1,fail[N],flg[N],n,S[N],Q[N],l,r,len[N],ans[N];
char w[N],s[N];
void insert(int x) {
	int p=1,i;
	for(i=1;w[i];i++) {
		int &k=ch[p][w[i]-'a'];
		if(!k) k=++cnt;
		p=k;
	}
	flg[p]=x;
}
void build() {
	int i,p;
	for(i=0;i<26;i++) ch[0][i]=1;
	Q[r++]=1;
	while(l<r) {
		p=Q[l++];
		for(i=0;i<26;i++) {
			if(ch[p][i]) fail[ch[p][i]]=ch[fail[p]][i],Q[r++]=ch[p][i];
			else ch[p][i]=ch[fail[p]][i];
		}
	}
}
int main() {
	scanf("%s%d",s+1,&n);
	int ls=strlen(s+1);
	int i;
	for(i=1;i<=n;i++) {
		scanf("%s",w+1); insert(i); len[i]=strlen(w+1);
	}
	build();
	// puts("FUCK");
	int p=1;
	for(i=1;i<=ls;i++) {
		p=ch[p][s[i]-'a'];
		S[++S[0]]=p; ans[S[0]]=s[i];
		if(flg[p]) S[0]-=len[flg[p]],p=S[S[0]];
	}
	for(i=1;i<=S[0];i++) {
		printf("%c",ans[i]);
	}
}