LeetCode题目笔记——520. 检测大写字母，从Python36ms到C++ 0ms

题目描述

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

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

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

示例 1：

输入：word = “USA”
输出：true

示例 2：

输入：word = “FlaG”
输出：false

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/detect-capital
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/detect-capital
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目难度——简单

方法一：按题目要求模拟

  题目要求要么全部是大写字母，要么全部是小写字母，要么只能用第一个字母是大写，其他全是小写，所以我们就可以遍历字符串，用一个counter来计数遇到的大写字母，最后判断这个counter和字符串长度的关系就可以了。

代码/Python

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        total = 0
        for i, w in enumerate(word):
            if w.isupper():
                total += 1

        if total == len(word):
            return True
        elif total == 0:
            return True
        elif total == 1 and word[0].isupper():
            return True
        else:
            return False

代码/C++

class Solution {
public:
    bool detectCapitalUse(string word) {
        int length = word.size(), i, total = 0;
        for(i = 0; i < length; i++){
            if (isupper(word[i])){
                total++;
            }
        }
        if (total == length){
            return true;
        }
        else if(total == 0){
            return true;
        }
        else if(total == 1 && isupper(word[0])){
            return true;
        }
        else{
            return false;
        }
    }
};

方法二：找规律

  如果第 1 个字母是大写，则其他字母必须都为大写或小写，即其他字母必须与第 2 个字母的大小写相同；而如果第一个字母是小写，则其他字母必须都为小写。

代码/Python

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        if len(word) >= 2 and word[0].islower() and word[1].isupper():
            return False
        
        return all(word[i].islower() == word[1].islower() for i in range(2, len(word)))

总结

  两种方法的时间复杂度都是O(N)，因为要遍历一遍字符串，空间负责度都是O(1)，因为只用到了常数空间。