leetcode 2020 第 189 场周赛 5413. 重新排列句子中的单词

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

句子的首字母大写
text 中的每个单词都用单个空格分隔。

请你重新排列 text 中的单词,使所有单词按其长度的升序排列。如果两个单词的长度相同,则保留其在原句子中的相对顺序。

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

输入:text = “Leetcode is cool”
输出:“Is cool leetcode”
解释:句子中共有 3 个单词,长度为 8 的 “Leetcode” ,长度为 2 的 “is” 以及长度为 4 的 “cool” 。
输出需要按单词的长度升序排列,新句子中的第一个单词首字母需要大写。
输入:text = “Keep calm and code on”
输出:“On and keep calm code”
解释:输出的排序情况如下:
“On” 2 个字母。
“and” 3 个字母。
“keep” 4 个字母,因为存在长度相同的其他单词,所以它们之间需要保留在原句子中的相对顺序。
“calm” 4 个字母。
“code” 4 个字母。

瞎模拟即可,结构体排个序,先按字符串长度排序,再按下标排序

#define MAXN ((int)1e5+7)
struct Node {
    int id;
    string str;
    bool operator < (const Node& no) const {
        return (str.length() == no.str.length()) ? id < no.id : (str.length() < no.str.length());
    }
} ;
class Solution {
public:
    string arrangeWords(string text) {
        text[0] = text[0]-'A' + 'a'; //开头大写转小写
        stringstream ss(text);
        string str;
        vector<Node> vec;
        int i = 1;
        while(ss >> str) {
            vec.push_back({i++, str});
        }
        sort(vec.begin(), vec.end());
        string ret;
        for(i=0; i<vec.size(); i++) {
            if(i == 0) { //开头大写
                vec[0].str[0] = vec[0].str[0]-'a' + 'A';
            }
            if(i) ret.push_back(' ');
            ret += vec[i].str;
        }
        return ret;
    }
};

你可能感兴趣的:(排序,leetcode)