LeetCode 151. 翻转字符串里的单词

题目描述

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

示例 1：

输入: "the sky is blue"
输出: "blue is sky the"

示例 2：

输入: "  hello world!  "
输出: "world! hello"
解释: 输入字符串可以在前面或者后面包含多余的空格，但是反转后的字符不能包括。

示例 3：

输入: "a good   example"
输出: "example good a"
解释: 如果两个单词间有多余的空格，将反转后单词间的空格减少到只含一个。

说明：

• 无空格字符构成一个单词。
• 输入字符串可以在前面或者后面包含多余的空格，但是反转后的字符不能包括。
• 如果两个单词间有多余的空格，将反转后单词间的空格减少到只含一个。

进阶：

请选用 C 语言的用户尝试使用 O(1) 额外空间复杂度的原地解法。

问题分析

此题分为三个步骤。1. 先将字符串中的多余空格都去掉，将字符串变为首尾无空格、单词中间只有一个空格的字符串。这里用一个next指针，初始值为0。从前往后遍历字符串，当遇到的是字符时，将next指针处置为该字符，然后将next加1，然后判断如果该字符的下一个位置是空格的话，将next指的位置设为空格，然后将next加1。当遍历结束后，先判断next如果为0，那么说明字符串中没有字符，所以返回空字符串；如果不为0，那么分为两种情况：字符串末尾是空格，那么接下来截取的字符串长度为前next-1；如果字符串末尾是字符，那么接下来截取的字符串长度为前next。下一步是将字符串前后翻转，再下一步是从前往后遍历字符串，遇到空格或者结束则将前一部分的单词翻转。最后结束返回结果。

代码实现

class Solution {
public:
    string reverseWords(string s) {
        int next = 0;
        for(int i = 0; i < s.size(); i++)
            if(s[i] != ' '){
                s[next++] = s[i];
                if(i < s.size() - 1 && s[i+1] == ' '){
                    s[next] = ' ';
                    next++;
                }
            }
        if(next == 0)
            return "";
        if(s.back() == ' ')
            s = s.substr(0, next - 1);
        else
            s = s.substr(0, next);
        reverse(s.begin(), s.end());
        int start = 0;
        for(int i = 0; i < s.size(); i++){
            if(i == s.size() - 1)
                reverse(s.begin() + start, s.end());
            if(s[i] == ' '){
                reverse(s.begin() + start, s.begin() + i);
                start = i + 1;
            }
        }
        return s;
    }
};