動機
日常一水題
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:])