LeetCode每日一题[C++]-2129.将标题首字母大写

题目描述

题目链接：. - 力扣（LeetCode）

给你一个字符串 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 ，所以其他单词首字母大写，剩余字母小写。

解题思路

我们顺序遍历 title 字符串，对于其中每个以空格为分界的单词，我们首先找出它的起始与末尾下标，判断它的长度以进行相应操作：

• 如果长度小于等于 2，则我们将该单词全部转化为小写；
• 如果长度大于 2，则我们将该单词首字母转化为大写，其余字母转化为小写。

最终，我们将转化后的字符串返回作为答案。

复杂度分析

时间复杂度：O(n)，其中 n 为 title的长度。即为遍历字符串完成操作的时间复杂度。

空间复杂度：O(1)，针对C++空间复杂度为O(1)

代码

#include <vector>
#include <iostream>
using namespace std;
class Solution
{
public:
    string capitalizeTitle(string title) {
        int n = title.size();
        int l = 0, r = 0;   // 单词左右边界（左闭右开）
        while (r < n) {
            while (r < n && title[r] != ' ') {
                ++r;
            }
            // 对于每个单词按要求处理
            if (r - l > 2) {
                title[l++] = toupper(title[l]);
            }
            while (l < r) {
                title[l++] = tolower(title[l]);
            }
            l = ++r;
        }
        return title;
    }

};
void printVector(vector<int> res)
{
    for (size_t i = 0; i < res.size(); i++)
    {
        cout << res[i] << " ";
    }
    cout << endl;
}
int main()
{
    Solution solution;
    string title = "capiTalIze tHe title";
    string res = solution.capitalizeTitle(title);
    cout << res << endl;

    title = "First leTTeR of EACH Word";
    res = solution.capitalizeTitle(title);
    cout << res << endl;

    title = "i lOve leetcode";
    res = solution.capitalizeTitle(title);
    cout << res << endl;
}