Leetcode1451 重新排列句子中的单词

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

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

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

输入：text = "Keep calm and code on"
输出："On and keep calm code"

实现代码

public String arrangeWords(String text) {
        String[] s = text.toLowerCase().split(" ");
        Arrays.sort(s, new Comparator<>(){
            @Override
            public int compare(String str1, String str2){
                if(str1.length() > str2.length())
                    return 1;
                else if(str1.length() < str2.length())
                    return -1;
                else 
                    return 0;
            }
           
        });
        s[0] = s[0].substring(0, 1).toUpperCase() + s[0].substring(1);
        return String.join(" ", s);


    }