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

문제 링크 (Level 1)

https://programmers.co.kr/learn/courses/30/lessons/12916?language=python3 

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

 

나의 코드 (Python)

def solution(s):

    p_cnt = 0
    y_cnt = 0

    for i in s.lower():
        if i == "p": 	# 문자열이 p일때
            p_cnt += 1
        elif i == "y": 	# 문자열이 y일때
            y_cnt += 1

    return True if p_cnt == y_cnt else False

 

1. 문자열에 대소문자를 구별하지 않기 위해 lower()를 사용하여 모두 소문자로 변경합니다.

2. p일 때는 p카운팅, y일 때는 y카운팅을 합니다

3. p카운팅, y카운팅이 같다면 True, 다르면 False를 리턴합니다.