[Python/Programmers/Level 1] 문자열 내 p와 y의 개수

출처 : programmers.co.kr/learn/courses/30/lessons/12916

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

* 이 포스팅은 Programmers 코딩 테스트 연습 Level 1 "문자열 내 p와 y의 개수" 문제풀이입니다.

   문제 원본은 링크를 클릭하시면 확인하실 수 있습니다.

# 문제 설명

• 대문자와 소문자가 섞여있는 문자열 s가 주어집니다.
• s에 "p"의 개수와 "y"의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution을 완성하세요.
• "p"와 "y" 모두 하나도 없는 경우는 항상 True를 리턴합니다.
• 단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다.
• 예를 들어 s가 "pPoooyY"면 true를 return하고 "Pyy"라면 false를 return 합니다.

# 제한사항

• 문자열 s의 길이 : 50 이하의 자연수
• 문자열 s는 알파벳으로만 이루어져 있습니다.

입출력 예

s	answer
"pPoooyY"	true
"Pyy"	false

# 코드 설명

def solution(s):
    answer = True
    
    count_p = s.count("p") + s.count("P")
    count_y = s.count("y") + s.count("Y")
    
    if count_p != count_y:
        answer = False
    
    return answer