【leetcode题解C++】541.反转字符串II and 151.反转字符串中的单词

541. 反转字符串II

给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。

• 如果剩余字符少于 k 个，则将剩余字符全部反转。
• 如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

示例 1：

输入：s = "abcdefg", k = 2
输出："bacdfeg"

示例 2：

输入：s = "abcd", k = 2
输出："bacd"

思路1：根据344.反转字符串，以及本题用到了STL，第一时间想到了reverse()库函数，但是在344题的题解中建议我们不要使用库函数，那么swap()。用一个tmp字符串累计存储2k个字符，处理完过后，加入ans字符串，接下来的话，就是一个很暴力的解法了。

需要注意的就是swap()中下标不要写错了。

代码实现1：

class Solution {
public:
    string reverseStr(string s, int k) {
        string ans;
        int p = 0;
        int round = 0;
        while (p < s.length()) {
            string tmp;
            while ((p - 2 * k * round) < 2 * k) {
                if (p < s.length()) {
                    tmp.push_back(s[p]);
                    ++p;
                } else {
                    break;
                }
            }
            ++round;
            if (tmp.length() < k) {
                for (int i = 0; i < tmp.length() - i; ++i) {
                    swap(tmp[i], tmp[tmp.length() - i - 1]);
                }
            }
            else if (k <= tmp.length() < 2*k) {
                for (int i = 0; i < k - i - 1; ++i) {
                    swap(tmp[i], tmp[k - i - 1]);
                }
            }
            else if (tmp.length() == 2*k) {
                for (int i = 0; i < k - i - 1; ++i) {
                    swap(tmp[i], tmp[k - i -1]);
                }
            }
            ans.append(tmp);
        }
        return ans;
    }
};

思路2：官方题解中使用了reverse()...代码量少了很多，这让作者很高兴啊hah...

代码实现2：

class Solution {
public:
    string reverseStr(string s, int k) {
        for (int i = 0; i < s.size(); i += (2 * k)) {
            // 1. 每隔 2k 个字符的前 k 个字符进行反转
            // 2. 剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符
            if (i + k <= s.size()) {
                reverse(s.begin() + i, s.begin() + i + k );
            } else {
                // 3. 剩余字符少于 k 个，则将剩余字符全部反转。
                reverse(s.begin() + i, s.end());
            }
        }
        return s;
    }
};

151. 反转字符串中的单词 

给你一个字符串 s ，请你反转字符串中 单词 的顺序。

单词 是由非空格字符组成的字符串。s 中使用至少一个空格将字符串中的 单词 分隔开。

返回 单词 顺序颠倒且 单词 之间用单个空格连接的结果字符串。

注意：输入字符串 s中可能会存在前导空格、尾随空格或者单词间的多个空格。返回的结果字符串中，单词间应当仅用单个空格分隔，且不包含任何额外的空格。

示例 1：

输入：the sky is blue输出：blue is sky the

示例 2：

输入：s = "  hello world  "
输出："world hello"
解释：反转后的字符串中不能存在前导空格和尾随空格。

示例 3：

输入：s = "a good   example"
输出："example good a"
解释：如果两个单词间有多余的空格，反转后的字符串需要将单词间的空格减少到仅有一个。

思路1：每个分割出来，反转，插入新的容器。（不提供代码了）

思路2：先去除多余的空格，整个字符串反转，再对每个单词反转，此处删除空格可能有些许疑难，但是我们用到27.移除元素的思路，用一个指针，遇到多余的空格就用后面的元素覆盖，知道把多余的空格都移到字符串的最后，再用resize()和指针指向的位置，来截去末尾的空格。反转的话，很简单，用swap就好了，注意先记录需要swap的起始位置和终止位置。

关键代码如下：

void removeExtraSpace(string &s) {
    int index = 0; //记录新字符串长度
    for(int i = 0; i < s.size(); ++i) {
        if(s[i] != ' ') {//如果不是空格，做一些处理；时空格的话就往后扫描
            if(slow != 0) s[slow++] = ' ';//第一个字符前面是不可能需要加空格的
            while(s[i] != ' ' && i < s.size()) {
                s[slow++] = s[i++];//录入字符
            }
        }
    }
}

代码实现2：

class Solution {
public:
    void removeExtraSpace(string& s) {
        int index = 0;
        for(int i = 0; i < s.size(); ++i) {
            if(s[i] != ' ') {
                if(index != 0) {
                    s[index++] = ' ';
                }
                while(i < s.size() && s[i] != ' ') {
                    s[index++] = s[i++];
                }
            }
        }
        s.resize(index);
    }
    
    void swapWords(string& s, int begin, int end) {
            for(int i = begin, j = end; i < j; ++i, --j) swap(s[i], s[j]);
        }

    string reverseWords(string s) {
        removeExtraSpace(s);
        swapWords(s, 0, s.size() - 1);
        int index = 0;
        for(int i = 0; i <= s.size(); ++i) {
            if (s[i] == ' ' || i == s.size()) {
                swapWords(s, index, i - 1);
                index = i + 1;
            }
        }
        return s;
    }
};