力扣151反转字符串里的单词(c++)

力扣151反转字符串里的单词(c++)

力扣题目链接

思路

按照力扣原题的进阶要求来书写代码，即不使用辅助空间，空间复杂度要求为O(1)。
不使用辅助空间，我们可以在原字符串上下功夫。

如果我们将整个字符串都反转一下，那么整个字符串的顺序就是倒序了，但单词本身也会是倒序，再把各个单词反转一下。
解题思路如下：
1.移除多余空格
2.将整个字符串反转
3.将每个单词反转

举个例子，源字符串为：" learn the algorithm well "
移除多余空格 : “learn the algorithm well”
字符串反转：“llew mhtirogla nrael”
单词反转：" well algorithm the learn "
这样我们就完成了翻转字符串里的单词。

移除空格我们采用双指针法去做，最后再resize（重新设置）一下字符串的大小，就可以做到O(n)的时间复杂度。

代码如下

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]);
        }
    }

    //去除字符串多余空格
    void removeExtraSpaces(string &s)
    {                 
        //去除所有空格并在相邻单词之间添加空格, 快慢指针。
        int slow = 0; 
        for (int i = 0; i < s.size(); ++i)
        { 
            if (s[i] != ' ')
            { //遇到非空格就处理，即删除所有空格。
                if (slow != 0)
                    s[slow++] = ' '; //手动控制空格，给单词之间添加空格。slow != 0说明不是第一个单词，需要在单词前添加空格。
                while (i < s.size() && s[i] != ' ')
                { //补上该单词，遇到空格说明单词结束。
                    s[slow++] = s[i++];
                }
            }
        }
        s.resize(slow); // slow的大小即为去除多余空格后的大小。
    }

    string reverseWords(string s)
    {
        removeExtraSpaces(s); //去除多余空格，保证单词之间之只有一个空格，且字符串首尾没空格。
        reverse(s, 0, s.size() - 1);
        int start = 0; // removeExtraSpaces后保证第一个单词的开始下标一定是0。
        for (int i = 0; i <= s.size(); ++i)
        {
            if (i == s.size() || s[i] == ' ')
            {                             //到达空格或者串尾，说明一个单词结束。进行翻转。
                reverse(s, start, i - 1); //翻转，注意是左闭右闭 []的翻转。
                start = i + 1;            //更新下一个单词的开始下标start
            }
        }
        return s;
    }
};

代码运行结果