Leetcode OJ: Reverse Words in a String

Reverse Words in a String

Given an input string, reverse the string word by word.

For example,
Given s = "the sky is blue",
return "blue is sky the".

单词反转，思路就是单词反转后再整句反转，只是加了一些要求，首尾的空格不能要，中间出现连续的空格都变成一个。

class Solution {
public:
    void reverseAll(string &s, int i, int j) {
        while(i < j) {
            if (s[i] != s[j])
                swap(s[i++], s[j--]);
            else {
                i++;
                j--;
            }
        }
    }
    void reverseWords(string &s) {
        size_t len = s.size();
        int pre = 0;
        int i = 0;
        // 清除开头的空格
        while (i < len && s[i] == ' ')
            i++;
        int begin = i++;
        int k = 0;
        // 把中间的空格合并
        while (i < len) {
            if (s[i - 1] == ' ' && s[i] == ' ') {
                k++;
            }else if (k > 0) {
                s[i - k] = s[i];
            }
            ++i;
        }
        len -= k;
        // 清除结尾的空格
        while (len > begin && s[len - 1] == ' ')
            len--;
        pre = begin;
        
        // 反转单词
        for (i = begin; i < len; ++i) {
            if (s[i] == ' ') {
                if (i - 1 > pre)
                    reverseAll(s, pre, i - 1);
                pre = i + 1;
            }
        }
        
        // 反转最后一个单词
        if (pre < len - 1)
            reverseAll(s, pre, len - 1);
        
        // 反转整句
        reverseAll(s, begin, len - 1);
        if (len < s.size())
            s.erase(len);
        if (begin > 0)
            s.erase(0, begin);
        
    }
};