代码随想录C++ Day7 | 344.反转字符串 541.反转字符串II 54. 替换数字（第八期模拟笔试）

344.反转字符串

 最直接的方法是创建一个同样大小的数组，从后往前将原数组的所有元素填入。但本题要求空间复杂度O(1)，因此必须在原数组上进行操作。

// 时间复杂度：O(n)
// 空间复杂度：O(1)
class Solution {
public:
    void reverseString(vector<char>& s) {
        // 左右双指针
        // 奇数个元素时，最中间元素不用管
        for (int i = 0, j = s.size() - 1; i < s.size() / 2; i++, j--) { 
            swap(s[i], s[j]);
        }
    }
};

541.反转字符串II

这道题只不过就是比上一道多了一些判断条件，因此只需要将这些条件写出来就行，交换的函数可以直接用库函数

// 时间复杂度：O(n)
// 空间复杂度：O(1)
class Solution {
public:
    string reverseStr(string s, int k) {
        int round = s.size() / (2 * k); // 完整操作几轮
        int res = s.size() % (2 * k); // 最后剩余字符数
        for (int i = 0; i < round; i++) {
            reverse(s.begin() + 2 * k * i, s.begin() + 2 * k * i + k);
        }
        if (res < k) {
            reverse(s.end() - res, s.end());
        }
        if (res >= k) {
            reverse(s.end() - res, s.end() - res + k);
        }
        return s;
    }
};

但这么写不够优雅，下边这样写更简洁，可以将情况1，2归为一类，这里自己实现reverse函数

// 时间复杂度：O(n)
// 空间复杂度：O(1)
class Solution {
public:
    void reverse(string &s, int start, int end) {
        for (int i = start, j = end; i < j; i++, j--) {
            swap(s[i], s[j]);
        }
    }

    string reverseStr(string s, int k) {
        for (int i = 0; i < s.size(); i += 2 * k) {
            // 1. 每隔 2k 个字符的前 k 个字符进行反转
            // 2. 剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符
            // 这里i已经指向了新一段的首个字符，因此，字符末尾索引是i+k-1
            if (i + k - 1 <= s.size() -1) { // 自然，两侧的-1可以消去
                reverse(s, i, i + k - 1); // 注意这里end参数传入的是i + k -1, 和库里的reverse函数不同
            } else {
                // 3. 剩余字符小于k，则反转所有剩余字符
                reverse(s, i, s.size() - 1);
            }
        }
        return s;
    }
};

54. 替换数字（第八期模拟笔试）

其实很多数组填充类的问题，其做法都是先预先给数组扩容带填充后的大小，然后在从后向前进行操作。

这么做有两个好处：

1. 不用申请新数组。(Java字符串不能修改，必须创建辅助空间)
2. 从后向前填充元素，避免了从前向后填充元素时，每次添加元素都要将添加元素之后的所有元素向后移动的问题。

 本题使用双指针法，i指向新长度的末尾，j指向旧长度的末尾。从后往前修改也不用担心i未遍历到的数字被j指针内容覆盖

#include <iostream>
#include <string>

using namespace std;

int main() {
    string s;
    while (cin >> s) {
        int sOldIndex = s.size() - 1;
        int count = 0; // 统计数字的个数
        for (int i = 0; i < s.size(); i++) {
            if (s[i] >= '0' && s[i] <= '9') {
                count++;
            }
        }
        // 扩充字符串s的大小，也就是将每个数字替换成"number"之后的大小
        s.resize(s.size() + count * 5);
        int sNewIndex = s.size() - 1;
        // 从后往前将数字替换为"number"
        while (sOldIndex >= 0) {
            if (s[sOldIndex] >= '0' && s[sOldIndex] <= '9') {
                s[sNewIndex--] = 'r';
                s[sNewIndex--] = 'e';
                s[sNewIndex--] = 'b';
                s[sNewIndex--] = 'm';
                s[sNewIndex--] = 'u';
                s[sNewIndex--] = 'n';
            } else {
                s[sNewIndex--] = s[sOldIndex];
            }
            sOldIndex--;
        }
        cout << s << endl;       
    }
}