Leetcode 520. Detect Capital

Seawolf

Given a word, you need to judge whether the usage of capitals in it is right or not.

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".
Otherwise, we define that this word doesn't use capitals in a right way.
 

Example 1:

Input: "USA"
Output: True
 

Example 2:

Input: "FlaG"
Output: False
 

Note: The input will be a non-empty word consisting of uppercase and lowercase latin letters

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        small = False
        count_B = 0
        count_S = 0
        if 97 <= ord(word[0]) <= 122:
            small = True
        else:
            small = False
        for i in range(1, len(word)):
            if small == True:
                if not 97 <= ord(word[i]) <= 122:
                    return False
                else:
                    count_S += 1
            else:
                if 97 <= ord(word[i]) <= 122:
                    count_S += 1
                elif 65 <= ord(word[i]) <= 90:
                    count_B += 1
        if small == True:
            return count_S == len(word) - 1
        else:
            return count_S == len(word) - 1 or count_B == len(word) - 1