[프로그래머스] 문자열 내 p와 y의 개수 (Python)

 

def solution(s):
    answer = True
    
    str = s.lower()
    p = str.count('p')
    y = str.count('y')
    
    if p != y:
        answer = False

    return answer

 

answer을 True로 초기화한 후, lower()을 통해 모두 소문자로 바꾼다.

count() 함수를 통해 p와 y의 개수를 찾고, True일 경우는 p == 0 and y == 0일 때, 그리고 두 개의 값이 같을 경우이므로 그 값이 다를 때만 체크를 한 번만 해 False로 바꿔준 후 리턴한다.