1. 问题描述:
「句子」是一个用空格分隔单词的字符串。给你一个满足下述格式的句子 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
2. 思路分析:
① 思路还是比较好理解的,我们需要注意题目中的两个规则即可,第一个是需要将对原字符串中的单词的按照长度由小到大进行排列,第二个是当单词长度一样的时候那么按照原来的位置进行排列,对于这个特点我们很容易想到使用map(可以使用Treemap因为利用Treemap可以对map中的键进行排序)来进行映射,将所有长度是一样的单词放置在一起,map中键为单词的长度,值为当前长度的单词有哪些,考虑到长度相同的单词位置不变那么值使用Deque<String>类型,这样利用队列先进先出的特点即可保证相对位置不变
② 还需要注意一个问题是单词的大小写问题,在一开始的时候我们可以先将句子中的大写的首字母转为小写,最后将句子中的首字母转为大写
3. 代码如下:
class Solution {
public String arrangeWords(String text) {
if (text == null || text.equals("")) return text;
text = text.trim();
/*一开始的时候就将其转为首字母小写*/
text = (char)(text.charAt(0) + 32) + text.substring(1);
String []words = text.split(" ");
TreeMap<Integer, Deque<String>> map = new TreeMap<>();
for (int i = 0; i < words.length; ++i){
int curlen = words[i].length();
Deque<String> cur = map.getOrDefault(curlen, new LinkedList<>());
cur.add(words[i]);
map.put(curlen, cur);
}
String res = "";
/*遍历map*/
for (Map.Entry<Integer, Deque<String>> cur : map.entrySet()){
Deque<String> deque = cur.getValue();
while (!deque.isEmpty()){
res += deque.poll() + " ";
}
}
/*将首字母转为大写*/
res = res.trim();
char c = res.charAt(0);
if (c >= 'a' && c <= 'z') c -= 32;
return c + res.substring(1);
}
}