動機

日常一水題

Problem

We define the usage of capitals in a word to be right when one of the following cases holds:

  • All letters in this word are capitals, like USA.
  • All letters in this word are not capitals, like leetcode.
  • Only the first letter in this word is capital, like Google.

Given a string word, return true if the usage of capitals in it is right.

 

Example 1:

Input: word = USAOutput: true

Example 2:

Input: word = FlaGOutput: false

 

Constraints:

  • 1 <= word.length <= 100
  • word consists of lowercase and uppercase English letters.

Sol

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        if word[0].isupper():
            return all(c.isupper() for c in word[1:]) or all(c.islower() for c in word[1:])
        else:
            return all(c.islower() for c in word[1:])