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

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

코딩테스트 연습 - 문자열 내 p와 y의 개수

for문 하나로 p와 y 갯수 구해서 비교해주면 됨. 

 

내 코드

def solution(s):
    answer = True
    p_cnt = 0
    y_cnt = 0
    for i in s:
        if i == 'p' or i == 'P':
            p_cnt += 1
        elif i == 'y' or i == 'Y':
            y_cnt += 1
    
    if p_cnt == y_cnt:
        return True
    else:
        return False

 

다른 사람 풀이

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

lower, count함수사용하면 한줄로 풀 수 있다.