LeetCode-in-Java

1451. Rearrange Words in a Sentence

Medium

Given a sentence text (A sentence is a string of space-separated words) in the following format:

Your task is to rearrange the words in text such that all words are rearranged in an increasing order of their lengths. If two words have the same length, arrange them in their original order.

Return the new text following the format shown above.

Example 1:

Input: text = “Leetcode is cool”

Output: “Is cool leetcode”

Explanation: There are 3 words, “Leetcode” of length 8, “is” of length 2 and “cool” of length 4. Output is ordered by length and the new first word starts with capital letter.

Example 2:

Input: text = “Keep calm and code on”

Output: “On and keep calm code”

Explanation: Output is ordered as follows:

“On” 2 letters.

“and” 3 letters.

“keep” 4 letters in case of tie order by position in original text.

“calm” 4 letters.

“code” 4 letters.

Example 3:

Input: text = “To be or not to be”

Output: “To be or to be not”

Constraints:

Solution

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

public class Solution {
    public String arrangeWords(String text) {
        TreeMap<Integer, List<String>> map = new TreeMap<>();
        String[] words = text.split(" ");
        for (String word : words) {
            int len = word.length();
            map.putIfAbsent(len, new ArrayList<>());

            map.get(len).add(word.toLowerCase());
        }
        StringBuilder sb = new StringBuilder();
        boolean first = true;
        for (Map.Entry<Integer, List<String>> len : map.entrySet()) {
            List<String> strings = len.getValue();
            for (String str : strings) {
                if (first) {
                    str = Character.toUpperCase(str.charAt(0)) + str.substring(1);
                    first = false;
                }
                sb.append(str).append(" ");
            }
        }
        return sb.substring(0, sb.length() - 1);
    }
}