Programmers / 문자열 내 p와 y의 개수 / Python

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: 문자열 내 p와 y의 개수

문제 사이트: https://programmers.co.kr/learn/courses/30/lessons/12916

---

나의 풀이

def solution(s):
    p_cnt = 0
    y_cnt = 0
    s = s.upper()
    for i in range(len(s)):
        if s[i] == 'P':
            p_cnt += 1
        elif s[i] == 'Y':
            y_cnt += 1
    if p_cnt == y_cnt:
        return True
    else:
        return False

 

---

count 를 이용해서도 풀 수 있을 거 같다