D. DS串应用--串替换

题目描述

给出主串、模式串、替换串，用KMP算法找出模式串在主串的位置，然后用替换串的字符替换掉模式串

本题只考虑一处替换的情况，如果你想做的完美一些，能够实现多处替换那

可能需要考虑模式串和替换串长度不一致的情况

输入

第一个输入t，表示有t个实例

第二行输入第1个实例的主串，第三行输入第1个实例的模式串，第四行输入第1个实例的替换串

以此类推

输出

第一行输出第1个实例的主串

第二行输出第1个实例的主串替换后结果，如果没有发生替换就输出主串原来的内容。

以此类推

样例查看模式

正常显示查看格式

输入样例1<-复制

3
aabbccdd
bb
ff
aaabbbccc
ddd
eee
abcdef
abc
ccccc
 

输出样例1

aabbccdd
aaffccdd
aaabbbccc
aaabbbccc
abcdef
cccccdef
 

#include<iostream>
#include<cstring>
using namespace std;

int* get_next(string str) {
	int len = (int)str.size();
	int j = 0, k = -1;
	int* next = new int[len+1];
	next[j] = k;
	while (k < len) {
		if (k == -1 || (str[j] == str[k])) {
			j++; k++;
			str[j] = k;
		}
		else {
			k = str[k];
		}
	}
	return next;
}

int KMPIndex(string m, string z) {
	int* next = get_next(z);
	int i = 0, j = 0;
	while (i < (int)m.size() && j < (int)z.size()) {
		if (j == -1 || m[i] == z[j]) {
			i++; j++;
		}
		else {
			j = next[j];
		}
	}
	if (j == (int)z.size()) {
		return i - j + 1;
	}
	else {
		return 0;
	}
}

int main() {
	int t;
	cin >> t;
	while (t--) {
		string m, z;
		string ti;
		cin >> m >> z>>ti;
		cout << m << endl;
		int index = KMPIndex(m, z);
		if (index == 0) {
			cout << m << endl;
		}
		else {
			for (int i = 0; i < index-1; i++) {
				cout << m[i];
			}
			for (int i = 0; i < (int)ti.size(); i++) {
				cout << ti[i];
			}
			for (int i = index + (int)z.size()-1; i < (int)m.size(); i++) {
				cout << m[i];
			}
			cout << endl;
		}
	}
	return 0;
}