LeetCode 520. Detect Capital

題目:

給定一個字，判斷大寫方式是否正確，有三種狀況是正確的，第一種是全部字都是大小，第二種是全部沒有大寫，第三種是只有第一個字母是大寫。

Example 1:

Input: "USA"
Output: True

Example 2:

Input: "FlaG"
Output: False

思路:

1. 計算大寫字母個數，接著用if判斷上述的三種情況

cap = 0
for i in word:
if i.isupper():
cap +=1

2.

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

Coding:

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        cap =0 
        for i in word:
            if i.isupper():
                cap +=1
        if cap == len(word):
            return True
        elif cap == 0:
            return True
        elif cap ==1 and word[0].isupper():
            return True
        else:    
            return False