题目描述

给定一个单词，你需要判断单词的大写使用是否正确。

我们定义，以下情况单词的大写用法是正确的：
1. 全部字母都是大写，比如”USA”。
2. 单词中所有字母都不是大写，比如”leetcode”。
3. 如果单词不只含有一个字母，只有首字母大写， 比如 “Google”。

否则，我们定义这个单词没有正确使用大写字母。

样例

输入: "USA"
输出: True

输入: "FlaG"
输出: False

注意

• 输入是由大写和小写拉丁字母组成的非空单词。

算法

(模拟) $O(n)$

1. 每个情况都分别写一个函数来判断是否符合这种情况。
2. 若都不符合，返回 false 即可。

时间复杂度

• 使用线性遍历判断是否符合情况，时间复杂度为 $O(n)$。

C++ 代码

class Solution {
public:
    bool all_capital(const string &word) {
        for (char c: word)
            if (! (c >= 'A' && c <= 'Z'))
                return false;

        return true;
    }

    bool all_not_capital(const string &word) {
        for (char c: word)
            if (! (c >= 'a' && c <= 'z'))
                return false;

        return true;
    }

    bool only_first_capital(const string &word) {
        if (word.length() < 2)
            return false;
        if (! (word[0] >= 'A' && word[0] <= 'Z'))
            return false;
        for (int i = 1; i < word.length(); i++)
            if (! (word[i] >= 'a' && word[i] <= 'z'))
                return false;
        return true;
    }

    bool detectCapitalUse(string word) {
        if (all_capital(word))
            return true;
        if (all_not_capital(word))
            return true;
        if (only_first_capital(word))
            return true;

        return false;
    }
};