洛谷：P4824 [USACO15FEB]Censoring S--KMP

来源：Censoring S

题目描述

原题来自：USACO 2015 Feb. Silver

给出两个字符串A和B ，每次从前往后找到A的一个子串B并将其删除，空缺位依次向前补齐，重复上述操作多次，直到A串中不含B串。输出最终的A串。

输入格式

第一行包含一个字符串 ，第二行包含一个字符串 。

输出格式

输出处理后的A串。

输入

whatthemomooofun
moo

输出

whatthefun

题意：给定串a，串b,  把a中所有的b删除，最后输出a串（删除一次后，后面的向前补齐）

题解： 这题只需要在原来kmp上稍加修改即可。

1. 先利用串a求出next数组
2. 在进行匹配过程中，循环每次都把当前字符加入答案数组
3. 当匹配成功，就将原来加入长度为m的串删除
4. 由于题目要求删除串b后，后面的串要向前补齐，所以每次还要记录j值便于回溯删除

 

code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e6+10;
char a[N],b[N],ans[N];
int n,m,l;
int p[N],q[N];
void get_() {
	p[0]=-1;
	for(int i=0,j=-1; i<m;) {
		if(j==-1||b[i]==b[j])p[++i]=++j;
		else j=p[j];
	}
}
void solve() {
	l=0;
	for(int i=0,j=0; i<n; i++) {
		ans[l++]=a[i];
		while(j&&a[i]!=b[j])j=p[j];
		if(a[i]==b[j])j++;
		q[l]=j;
		if(j==m) {
			l-=m;
			j=q[l];
		}
	}
}
int main() {
	scanf("%s%s",a,b);
	n=strlen(a);
	m=strlen(b);
	get_();
	solve();
	ans[l]='\0';
	printf("%s",ans);
	return 0;
}