151. 反转字符串中的单词

题目描述

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

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

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

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

示例 1：

输入：s = "the sky is blue"
输出："blue is sky the"

示例 2：

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

示例 3：

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

提示：

• 1 <= s.length <= 104
• s 包含英文大小写字母、数字和空格 ' '
• s 中 至少存在一个 单词

**进阶：**如果字符串在你使用的编程语言中是一种可变数据类型，请尝试使用 O(1) 额外空间复杂度的 原地 解法。

解答

class Solution {
public:
    string reverseWords(string s) {
        // 1.去掉多余空格
        removeExtraSpace(s);

        // 2.整体反转
        reverse(s, 0, s.size() - 1);
        int start = 0; // 第一个单词的起始位置
        // 3.反转每个单词
        for(int i = 0; i <= s.size(); ++i)
        {
            if(i == s.size() || s[i] == ' ')
            {
                reverse(s, start, i - 1);
                start = i + 1; // 下一个单词的起始位置
            }
        }
        return s;
    }

    void removeExtraSpace(string &s)
    {
        int slow = 0;
        for(int i = 0; i < s.size(); ++i)
        {
            // 这个逻辑会跳过字符串首尾空格
            // 对每个单词处理
            if(s[i] != ' ')
            {
                if(slow != 0) s[slow++] = ' '; // 从第二个单词开始之前都要加空格
                while(i < s.size() && s[i] != ' ')
                {
                    // 把单词往前面空格处移动
                    s[slow++] = s[i++];
                }
            }
        }
        s.resize(slow);
    }
    
    void reverse(string &s, int start, int end)
    {
        while(start < end)
        {
            swap(s[start], s[end]);
            start++; 
            end--;
        }
    }
};