작심삼일

[LeetCode] 520 | Detect Capital | Python 본문

스터디/코테

[LeetCode] 520 | Detect Capital | Python

yun_s 2023. 1. 2. 09:13
728x90
반응형

문제 링크: https://leetcode.com/problems/detect-capital/description/


문제

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.


조건

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

내 풀이

capitalize()는 첫글자는 대문자로, 다른 글자들은 소문자로 바꿔주는 함수이다.

만약 주어진 word가 첫글자만 대문자라면 word == word.capitalize()가 성립할 것이고, word가 모두 대문자라면 word.isupper()가, word가 모두 소문자라면 word.islower()가 성립할 것이다.

따라서 위에서 언급한 경우 중 하나를 만족하면 True를 리턴하고 그렇지 않다면 False를 리턴하도록 한다.


코드

class Solution:
    def detectCapitalUse(self, word: str) -> bool:
        return True if word.capitalize() == word or word.isupper() or word.islower() else False
728x90
반응형
Comments