【LeetCode】151. 翻转字符串里的单词 结题报告 (C++)

原题地址：https://leetcode-cn.com/problems/reverse-words-in-a-string/description/

题目描述：

给定一个字符串，逐个翻转字符串中的每个单词。

示例:  

输入: "the sky is blue",
输出: "blue is sky the".
说明:

无空格字符构成一个单词。
输入字符串可以在前面或者后面包含多余的空格，但是反转后的字符不能包括。
如果两个单词间有多余的空格，将反转后单词间的空格减少到只含一个。
进阶: 请选用C语言的用户尝试使用 O(1) 空间复杂度的原地解法。

 

解题方案：

这题看似简单，却花了我大量的时间找bug。。主要是这几个问题：

• 函数str.erase(it)，每次删除之后，it会自动加一，移到原先删除节点的下一个节点。
• string类型内部反转，reverse(s.begin(), s.end())。
• 字符串结尾处需要额外处理，比较容易出错。

代码：

class Solution {
public:
    void reverseWords(string &s) {
        if(s.length() == 0)
            return ;
    
        reverse(s.begin(), s.end());
        int flag = 0;
        auto it = s.begin(), pre = s.begin();
        while(*it == ' ' && it != s.end())
            s.erase(it);
        while(it != s.end()){
            if(*it == ' ' && flag == 1){
                s.erase(it);
                continue;
            }
            else if(*it == ' ' && flag == 0){
                flag = 1;
                reverse(pre, it);
            }
            else if(flag == 1){
                pre = it;
                flag = 0;
            }
            it ++;
        }
        if(flag == 0)
            reverse(pre, it);
        else
            s.erase(s.end() - 1);
    }
};