2129. 将标题首字母大写

给你一个字符串 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 由单个空格隔开的单词组成,且不含有任何前导或后缀空格。
  • 每个单词由大写和小写英文字母组成,且都是 非空 的。

题解:

class Solution {
public:
    string capitalizeTitle(string title) {
        int flag = 0;//用来记录每次遇到空格后的那一位
        for(int i = 0;i<title.size();i++){
            title[title.size()-1] = tolower(title[title.size()-1]);
            if(i==0){
                title[i] = toupper(title[i]);
                 title[title.size()-1] = tolower(title[title.size()-1]);
            }else if(title[i-1]==' '){
                if(i-2-flag<2){//判断一个单词是否超过2的长度的大小
                    title[flag] = tolower(title[flag]);
                }
                flag = i;
                title[i] = toupper(title[i]);
                title[title.size()-1] = tolower(title[title.size()-1]);
            }else if(i==title.size()-1&&i-flag+1<=2){
                if(i!=flag)title[i] = tolower(title[i]);
                title[flag] = tolower(title[flag]);
            }
            else{
                title[i] = tolower(title[i]);
            }
        }
        return title;
    }
};