520. Detect Capital

📋 Đề Bài

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 = "USA"
Output: true

Example 2:

Input: word = "FlaG"
Output: false

Constraints:

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

💻 Lời Giải

Python 0520-detect-capital.py

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        if word.upper() == word or word.lower() == word:
            return True
        
        n = len(word)
        if n >= 1 and word[0].upper() != word[0]:
            return False
        
        for i in range(1, n):
            if word[i].lower() != word[i]:
                return False
                
        return True