LeetCode刷题记录——第520题（检测大写字母）

题目描述

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

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

全部字母都是大写，比如"USA"。
单词中所有字母都不是大写，比如"leetcode"。
如果单词不只含有一个字母，只有首字母大写， 比如 “Google”。
否则，我们定义这个单词没有正确使用大写字母。

示例 1:

输入: “USA”
输出: True

示例 2:

输入: “FlaG”
输出: False
注意: 输入是由大写和小写拉丁字母组成的非空单词。

思路分析

• 如果word长度为1，那么肯定满足条件
• 如果第一个字符为小写，那么之后的字符需要全部小写才能满足条件
• 如果第一个字符为大写，之后之后的字符全为大写或者全为小写才能满足条件
• 通过if语句来确定流程

代码示例

class Solution(object):
    def detectCapitalUse(self, word):
        """
        :type word: str
        :rtype: bool
        """
        if len(word) == 1:
            return True
        elif word[0].islower():
            return word[1:].islower()
        elif word[0].isupper() and word[1].islower():
            return word[1:].islower()
        elif word[0].isupper() and word[1].isupper():
            return word[1:].isupper()