【数据结构OJ】DS串应用--串替换

Description

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

本题只考虑一处替换的情况，如果你想做的完美一些，能够实现多处替换那可能需要考虑模式串和替换串长度不一致的情况

Input

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

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

以此类推

Output

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

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

以此类推

Sample

samplein

4
aabbccdd
bb
ff
aaabbbccc
ddd
eee
abcdef
abc
ccccc
abcdef
abc
c

sampleout

aabbccdd
aaffccdd
aaabbbccc
aaabbbccc
abcdef
cccccdef
abcdef
cdef

分析

题目主要依赖KMP算法实现，可参考【数据结构OJ】A : DS串应用--KMP算法

然后仅做一点改变在KMPFind函数中（在前边返回值是从1开始计算，这里直接应用需要从0开始，无需加1）（这里看不懂没关系，不重要）

在main主函数中用到自带的replace函数即可。

AC代码

#include<iostream>
#include<string>
using namespace std;
class myString
{
private:
    string mainstr; // 串
    int size;       // 串长度
    void GetNext(string p, int next[]);
    int KMPFind(string p, int next[]);//我删掉了pos这个参数，作用不大
public:
    myString();
    ~myString();
    void SetVal(string sp);
    int KMPFindSubstr(string p);//这里也删掉了pos参数
};
myString::myString()
{
    size = 0;
    mainstr = "";
}
myString::~myString()
{
    size = 0;
    mainstr = "";
}
void myString::SetVal(string sp)
{
    mainstr = "";
    mainstr.assign(sp);
    size = mainstr.length();
}
int myString::KMPFindSubstr(string p)
{
    int i;
    int L = p.length();
    int* next = new int[L+1];
    GetNext(p, next);
    int v = -1;
    v = KMPFind(p, next);
    delete[]next;
    return v;
}

int myString::KMPFind(string p, int next[]) {
    int i=0, j=0;//i在主串循环，j在子串循环；
    int plen = p.length();
    int mlen = mainstr.length();
    while (i < mlen && j < plen) {
        if ((j == -1) || (mainstr[i] == p[j])) {
            ++i; ++j;;
        }
        else
            j = next[j];
    }
    if (j >= plen) {
        return i - p.length();//返回当前主串位置；这里做了修改，无需加1
    }
    else
        return -1;//没有匹配
}

void myString::GetNext(string p, int next[]) {
    next[0] = -1;
    int j = -1, k = 0;
    while (k < p.length()) {
        if (j == -1 || p[k] == p[j]) {// P[k]表示后缀的单个字符，P[j]表示前缀的单个字符
            ++k; ++j; next[k] = j;
        }
        else {
            j = next[j];
        }
    }
}

int main() {
    int t;
    cin >> t;
    while (t--)
    {
        myString str;
        string mainstr, substr1,substr2;
        cin >> mainstr >> substr1 >> substr2;
        cout << mainstr << endl;
        str.SetVal(mainstr);
        int ans;
        ans = str.KMPFindSubstr(substr1);
        if (ans == -1) {//无法匹配
            cout << mainstr << endl;
        }
        else {//匹配到则返回了匹配的位置进行替换
            mainstr.replace(ans, substr1.length(), substr2);
            cout << mainstr << endl;
        }

    }
    return 0;
}