[프로그래머스] Lv1 - 문자열 내 p와 y의 개수

문제

프로그래머스 연습문제

• Lv 1. 문자열 내 p와 y의 개수 (Python)
  https://school.programmers.co.kr/learn/courses/30/lessons/12916

코드 구현

def solution(s):
    answer = True
    
    num_p = 0
    num_y = 0
    
    for i in s:
        if(i == 'p' or i == 'P'):
            num_p += 1
        if(i == 'y' or i == 'Y'):
            num_y += 1
    
    if(num_p != num_y):
        return False

    return True

풀이

• 대소문자 구분없이 p나 y의 개수를 세서 값이 같은지 다른지 비교하면 된다.
• 더 간단하게 하려면 lower() 나 upper()를 사용하여 모두 소문자 혹은 대문자로 변경한 뒤, 값을 비교했을 것 같다.

What I learned

깔끔코드

lower()를 이용해 소문자로 통일시킨 뒤 → count()를 통해 갯수가 같은지 아닌지 리턴

def solution(s):
    return s.lower().count('p') == s.lower().count('y')