2017西安交大ACM小学期 敏感词汇[AC自动机]

敏感词汇

发布时间: 2017年7月5日 00:23   最后更新: 2017年7月6日 14:40   时间限制: 1500ms   内存限制: 128M

描述

我们知道，在进行聊天时，有些词汇是敏感词汇，含有敏感词汇的内容是不允许被发送的。现在给定m个敏感词汇，并给定一段文本，请将所有敏感词汇都用星号替换掉。

输入

包含多组数据。
每组数据第一行一个正整数 m ，表示有 m 个敏感词汇。
接下来 m 行每行一个敏感词汇，敏感词汇仅包含小写字母。
最后一行为文本，仅包含小写字母。
每组数据保证敏感词汇总长度不超过 106 ，文本不超过 106 。
总字符输入量不超过 107 。

输出

对每组数据，输出一行一个字符串，表示敏感词汇都用星号替换掉后的文本。

样例输入1  复制

3
naive
simple
glasses
glassesimplenaiveexcited

样例输出1

*****************excited

这是一个比较裸的AC自动机的题

这里要注意几点：

1.不能边检测边覆盖，不然肯定会超时，所以我们用一个replace数组记录下从当前这个位置应该向前覆盖*的个数

2.在检测到下一个字符，并且往前回溯的时候，只要搜索到第一个有效模式即可，因为剩下的模式长度只能更小，我们覆盖大模式的时候，肯定就可以覆盖掉小的模式

3.最后覆盖的时候，从后往前扫描replace数组，可以在O(n)时间内完成覆盖

代码：

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 1e6+7;
int replace[MAXN];
#define LETTER 26
struct Trie{
    int num, fail,match;
    int next[LETTER];
    int dep;
}pool[MAXN];
Trie* const trie = pool + 1;
int cnt;
void init(){
    cnt = 0;
    memset(pool, 0, 2 * sizeof(Trie));
    trie[0].fail = -1;
}
inline int convert(char c){
	return c - 'a';
}
void build()
{
  queue<int> q; q.push(0);
  while (!q.empty()){
    int t = q.front(); q.pop();
    for (int i = 0; i < LETTER; i++){
      int &cur = trie[t].next[i];
      if (cur){
        q.push(cur);
        trie[cur].fail = trie[trie[t].fail].next[i];
        trie[cur].match = trie[cur].num ? cur :trie[trie[cur].fail].match;
      }
      else cur = trie[trie[t].fail].next[i];
    }
  }
}
int search(char *s)
{
    int ret = 0, cur = 0;
    for (int i = 0; s[i]; i++){
        cur = trie[cur].next[convert(s[i])];
        for (int temp = trie[cur].match; temp;temp = trie[trie[temp].fail].match){
           	//ret += trie[temp].num;
           	if(trie[temp].num){
           		replace[i] = trie[temp].dep;
           		break;
           	}
            //trie[temp].num = 0;
		}
            
    }
    return ret;
}
void insert(char s[]){
	int cur = 0;
	for(int i = 0;s[i];i++){
		int &pos = trie[cur].next[convert(s[i])];
		if(!pos){
			pos = ++cnt;
			memset(&trie[cnt],0,sizeof(Trie));
		}
		trie[pos].dep = trie[cur].dep + 1;
		cur = pos;
	}
	trie[cur].num ++;
}

char pat[MAXN];
char str[MAXN];
int main(){
	int m;
	while(~scanf("%d",&m)){
		init();
		memset(replace,0,sizeof(replace));
		while(m--){
			scanf(" %s",pat);
			insert(pat);
		}
		build();
		scanf(" %s",str);
		int ans = search(str);
		//printf("%d\n",ans);
		int len = strlen(str);
		int cnt = 0;
		for(int i = len-1;i >= 0;i--){
			cnt = max(cnt,replace[i]);
			if(cnt){
				str[i] = '*';
				cnt--;
			}
		}
		puts(str);
	}
}