题目链接: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;
}