2129. Capitalize the Title-将标题首字母大写

给你一个字符串title，它由单个空格连接一个或多个单词组成，每个单词都只包含英文字母。请你按以下规则将每个单词的首字母大写：

• 如果单词的长度为 1 或者 2 ，所有字母变成小写。
• 否则，将单词首字母大写，剩余字母变成小写。

请你返回大写后的title。

示例 1：

输入：title = "capiTalIze tHe titLe"
输出："Capitalize The Title"
解释：
由于所有单词的长度都至少为 3 ，将每个单词首字母大写，剩余字母变为小写。

示例 2：

输入：title = "First leTTeR of EACH Word"
输出："First Letter of Each Word"
解释：
单词 "of" 长度为 2 ，所以它保持完全小写。
其他单词长度都至少为 3 ，所以其他单词首字母大写，剩余字母小写。

示例 3：

输入：title = "i lOve leetcode"
输出："i Love Leetcode"
解释：
单词 "i" 长度为 1 ，所以它保留小写。
其他单词长度都至少为 3 ，所以其他单词首字母大写，剩余字母小写。

提示：

• 1 <= title.length <= 100
• title由单个空格隔开的单词组成，且不含有任何前导或后缀空格。
• 每个单词由大写和小写英文字母组成，且都是 非空 的。

Python:

class Solution:
    def capitalizeTitle(self, title: str) -> str:
        arr = title.split()
        for i in range(len(arr)):
            if len(arr[i]) == 1 or len(arr[i]) == 2:
                arr[i] = arr[i].lower()
            else:
                arr[i] = arr[i][0].upper()+arr[i][1:].lower()
        return ' '.join(arr)

Java:

class Solution {
    public String capitalizeTitle(String title) {
        String[] arr = title.split(" ");
        for(int i = 0; i < arr.length; i++)
        {
            if(arr[i].length() == 1 || arr[i].length() == 2)
            {
                arr[i] = arr[i].toLowerCase();
            }
            else
            {
                arr[i] = arr[i].substring(0, 1).toUpperCase()+arr[i].substring(1).toLowerCase();
            }
        }
        return String.join(" ", arr);

    }
}