프로그래머스 - 문자열 내 p와 y의 개수 (파이썬3 Python3)

https://school.programmers.co.kr/learn/courses/30/lessons/12916

프로그래머스

 

 

문자열을 모두 소문자로 바꿔준 뒤, 

p와 y의 갯수를 비교해서 같으면 true, 다르면 false를 리턴해주면됩니다. 

 

def solution(s):
    answer = True
    
    s = s.lower()
    p = 0
    y = 0

    s = list(s)

    for i in range(len(s)):
        if s[i] == 'p':
            p += 1
        elif s[i] == 'y':
            y += 1
    
    if p != y:
        return False
    else:
        pass

    return True