문자열 내 p와 y의 개수

나의 풀이 

def solution(s):
    answer = True
    countP = 0
    countY = 0
    for s in s:
        if s == chr(80) or s == chr(112):
            countP += 1
        elif s == chr(89) or s == chr(121):
            countY += 1

    if countP == countY: 
        return True
    else: 
        return False
    return True

s = "pPoooyY" 
n = "pyY"


solution(s)
solution(n)

가볍지만 길다.

 

 

 

다른 사람 풀이 1

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

모두 소문자로 바꾸고 카운트 함수를 사용했다. 대소문자에서 불편함을 길게 가져갔었다면.....