[题解]剑指 Offer 58 - I. 翻转单词顺序(C++)

题目

输入一个英文句子,翻转句子中单词的顺序,但单词内字符的顺序不变。为简单起见,标点符号和普通字母一样处理。例如输入字符串"I am a student. ",则输出"student. a am I"。

示例 1:

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

示例 2:

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

示例 3:

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

说明:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/fan-zhuan-dan-ci-shun-xu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

要翻转单词的顺序同时保持单词内字符的顺序,很自然地想到利用栈先进后出的特性。遍历s把单词逐个入栈,然后抛出单词直到栈空,记得在首个单词以外的单词前面加上空格。
时间复杂度O(n),空间复杂度O(n)。

代码

class Solution {
public:
    string reverseWords(string s) {
        stack<string> stk;
        for(int left = 0; left < s.size(); ++left)
        {
            if(s[left] != ' ')
            {
                int right = left;
                while(right < s.size() && s[right] != ' ') right++;
                string tmp(s.begin() + left, s.begin() + right);
                stk.push(tmp);
                left = right;
            }
        }
                        
        string ans = "";
        while(!stk.empty())
        {
            if(!ans.empty()) ans += ' ';
            ans += stk.top();
            stk.pop();
        }
        return ans;
    }
};

改进

用栈解决需要额外O(n)的空间,能不能直接在传入的s上原地修改?利用c++的翻转函数reverse(也可以自己实现)可以做到,先翻转整个字符串,然后用双指针,write指向要写入的位置,read指向读到单词的位置;read读到单词时,用wordend向后遍历到空格或者字符串结束,把wordend遍历到的字符写到write指向的位置,然后再在新字符串离翻转新写进来的这个单词。注意每次写入单词前除了首个单词外要加上空格。
时间复杂度O(n),空间复杂度O(1)。

代码

class Solution {
public:
    string reverseWords(string s) {
        reverse(s.begin(), s.end());
        int write = 0, n = s.size();
        for(int read = 0; read < n; ++read)
        {
            if(s[read] != ' ')
            {
                if(write != 0) s[write++] = ' ';
                int wordend = read;
                while(wordend < n && s[wordend] != ' ') s[write++] = s[wordend++];
                reverse(s.begin() + write - (wordend - read), s.begin() + write);
                read = wordend;
            }
        }
        s.erase(s.begin() + write, s.end());
        return s;
    }
};
上一篇:LeetCode-字符串-58. 最后一个单词的长度


下一篇:58 动态内存分配的实现(上)