[Codeup] 问题 D: 单词替换

先把题目地址贴出来：(http://codeup.cn/problem.php?cid=100000580&pid=3)

输入一个字符串，以回车结束（字符串长度<=100）。该字符串由若干个单词组成，单词之间用一个空格隔开，所有单词区分大小写。现需要将其中的某个单词替换成另一个单词，并输出替换之后的字符串。

输入格式

多组数据。每组数据输入包括3行，

第1行是包含多个单词的字符串 s，

第2行是待替换的单词a，(长度<=100)

第3行是a将被替换的单词b。(长度<=100)

s, a, b 最前面和最后面都没有空格。

输出格式

每个测试数据输出只有 1 行，

将s中所有单词a替换成b之后的字符串。

样例

Input

I love Tian Qin
I
You

Output

You love Tian Qin

题解

代码如下：

find + Replace ：

#include<bits/stdc++.h>
using namespace std;
 
int main() {
    string aa, sub, New;
    while(getline(cin, aa))
    {
        cin >> sub >> New;
        getchar();
        int t = aa.find(sub);
        while(t != string::npos){
            aa = aa.replace(aa.begin()+t, aa.begin()+t+sub.size(), New);
            t = aa.find(sub);
        }
        cout << aa << endl;
    }
    return 0;
}

很不幸，测试的字符串由于太长，replace带来的开销太大。只能考虑换种解法。

于是想到，我们不一定非要将字符串进行替换，只要能正确输出即可。

所以：考虑以母串为aa，被替换子串sub，替换子串new。对母串进行遍历，只有两种情况：

1. 到了其中sub子串的位置，以样例中的输入为例，遍历到 I 的时候即满足，输出替换后的子串New，并且让遍历指针i后移一定的位数，相当于跳过母串中这里的sub。
2. 未到sub子串的位置，直接输出原串该位，i++。

于是有以下代码：

修改后的：

#include<bits/stdc++.h>
using namespace std;

int main() {
	string aa, sub, New;
	while(getline(cin, aa))
    {
        cin >> sub >> New;
        getchar(); //清除缓存区的'\n'
        for(int i = 0; i < aa.size(); )
        {
            if(i != aa.find(sub, i)) // 从当前位置开始查，若未遍历到欲替换的子串
                cout << aa[i++]; // 直接输出并让i++，输出原串
            else{ //到了该替换的地方，先输出新串，再完成i滑动
                cout << New;
                i += sub.size();
            }
        }
        cout << endl;
    }
	return 0;
}

Note

这也算是字符串类型题的一个常见思路吧。增删之类的操作会给线性表带来不小的开销，如果题目卡数据的话很容易被卡住。此时如果能不对字符串本身进行修改，换种方式正确输出答案，往往可以很好地解决问题。