관리 메뉴

솜씨좋은장씨

[leetCode] 520. Detect Capital (Python) 본문

Programming/코딩 1일 1문제

[leetCode] 520. Detect Capital (Python)

솜씨좋은장씨 2021. 1. 15. 20:46
728x90
반응형

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:

  1. All letters in this word are capitals, like "USA".
  2. All letters in this word are not capitals, like "leetcode".
  3. 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.

 

Solution

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        upper_word = word.upper()
        lower_word = word.lower()
        
        answer = False
        
        if upper_word == word or lower_word == word:
            answer = True
            
        if len(word) > 0 and (word[0] == upper_word[0] and word[1:] == lower_word[1:]):
            answer = True
            
        return answer

 

SOMJANG/CODINGTEST_PRACTICE

1일 1문제 since 2020.02.07. Contribute to SOMJANG/CODINGTEST_PRACTICE development by creating an account on GitHub.

github.com

728x90
반응형
0 Comments
댓글쓰기 폼