Leetcode 5413：重新排列句子中的单词

题目描述

「句子」是一个用空格分隔单词的字符串。给你一个满足下述格式的句子 text :

句子的首字母大写
text 中的每个单词都用单个空格分隔。
请你重新排列 text 中的单词，使所有单词按其长度的升序排列。如果两个单词的长度相同，则保留其在原句子中的相对顺序。

请同样按上述格式返回新的句子。

 

示例 1：

输入：text = "Leetcode is cool"
输出："Is cool leetcode"
解释：句子中共有 3 个单词，长度为 8 的 "Leetcode" ，长度为 2 的 "is" 以及长度为 4 的 "cool" 。
输出需要按单词的长度升序排列，新句子中的第一个单词首字母需要大写。
示例 2：

输入：text = "Keep calm and code on"
输出："On and keep calm code"
解释：输出的排序情况如下：
"On" 2 个字母。
"and" 3 个字母。
"keep" 4 个字母，因为存在长度相同的其他单词，所以它们之间需要保留在原句子中的相对顺序。
"calm" 4 个字母。
"code" 4 个字母。
示例 3：

输入：text = "To be or not to be"
输出："To be or to be not"
 

提示：

text 以大写字母开头，然后包含若干小写字母以及单词间的单个空格。
1 <= text.length <= 10^5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rearrange-words-in-a-sentence
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    string arrangeWords(string text) {
        string ans = "";
        int idx = 0;
        vector<pair<string,int>> vect;
        string::size_type pre = 0, post = 0;
        post = text.find_first_of(" ");
        while(pre < text.length()){
            string tmp = text.substr(pre, post-pre);
            if(tmp[0] >= 'A' && tmp[0] <= 'Z') tmp[0] += 32;
            vect.push_back(make_pair(tmp, idx));
            pre = post + 1;
            idx++;
            post = text.find_first_of(" ", pre);
            if(post == string::npos) post = text.length();
        }
        sort(vect.begin(), vect.end(), [](pair<string, int> &a, pair<string, int> &b){
            if(a.first.length() == b.first.length()) return a.second < b.second;
            return a.first.length() < b.first.length();
        });
        if(vect.size() >= 1) vect[0].first[0] -= 32;
        for(auto it : vect) ans += it.first + " ";
        if(ans.length() > 0) ans.pop_back();
        return ans;
    }
};