题目链接:https://ac.nowcoder.com/acm/contest/372/C
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
另一天,一只可爱的围着围巾的肥企鹅在路上摇摇晃晃地走着,遇上了迎面走来的打着饱嗝的PM6。小企鹅预感不妙,这不就是最近有名的恶人PM6么!吓得立刻扭头就想跑。
PM6:“小火汁,站住!我不吃你(谁叫你是保护动物)。我这有一道简单题,如果你答对了,我就给你吃鱼肉,如果你答错了,就免费帮我充游戏币!”
企鹅:“_(:3J∠)_(默默摘掉围巾)”
PM6:“我给你一个文本串 S ,再给你两个串A、B,你要将文本串中的 A 都转换成 B ,转换后的字符不再参与转换,输出最终的文本串。”
求求你救救企鹅!
输入描述
第一行输入一个文本串 S 。
第二行输入字符串 A 。
第三行输入字符串 B 。
|S|为S的长度,|A|为A的长度,|B|为B的长度,所有字符都是小写字母,保证 |A| <= |S| 。
对于50%的数据:1<= |A|、|B|、|S| <=1000
对于100%的数据:1<= |A|、|B|、|S| <=1000000
输出描述
只有一行,输出转换后的文本串。
输入
abababcd
ab
cd
输出
cdcdcdcd
解题思路
KMP,求出B串在A串出现的位置,输出的时候遇到记录的位置时输出C串就行了。
#include <bits/stdc++.h>
using namespace std;
int nex[1000005], num;
char a[1000005], b[1000005], c[1000005];
int lena, lenb, lenc;
void get_next() {
int i = 0, j = -1;
nex[0] = -1;
while (i < lenb) {
if (~j && b[i] != b[j])
j = nex[j];
else {
i++;
j++;
if (b[i] != b[j])
nex[i] = j;
else nex[i] = nex[j];
}
}
}
void index_KMP() {
int i, j;
num = i = j = 0;
while (i < lena) {
if (~j && a[i] != b[j])
j = nex[j];
else {
i++;
j++;
}
if (j == lenb) {
num++;
for (int k = i - lenb; k < i; k++)
a[k] = 0;
j = nex[j];
}
}
}
int main() {
while (~scanf("%s%s%s", a, b, c)) {
lena = strlen(a);
lenb = strlen(b);
lenc = strlen(c);
get_next();
index_KMP();
for (int i = 0; i < lena; i++) {
if (!a[i] && num) {
num--;
i += lenb - 1;
printf("%s", c);
}
else printf("%c", a[i]);
}
printf("\n");
}
return 0;
}