520. 检测大写字母-爱代码爱编程

520. 检测大写字母

题目

我们定义，在以下情况时，单词的大写用法是正确的：

• 全部字母都是大写，比如 "USA" 。
• 单词中所有字母都不是大写，比如 "leetcode" 。
• 如果单词不只含有一个字母，只有首字母大写， 比如 "Google" 。

给你一个字符串 word 。如果大写用法正确，返回 true ；否则，返回 false 。

提示：

• 1 <= word.length <= 100
• word 由小写和大写英文字母组成

示例

示例 1：

输入：word = "USA"
输出：true

示例 2：

输入：word = "FlaG"
输出：false

题解

统计一下字符串中的大写字母数量，如果为0或等于字母数量说明只有小写或大写字母，返回true。如果为1，并且这个唯一的大写字母是首字母也返回true，其余情况返回false。

class Solution {
public:
    bool detectCapitalUse(string word) {
        int number_of_big = 0;
        int number = word.length();
        for(char ch: word){
            if('A' <= ch && ch <= 'Z') {
                number_of_big++;
            }
        }
        return number_of_big == 0 || number_of_big == number || (number_of_big == 1 && 'A' <= word[0] && word[0] <= 'Z');
    }
};