programming

프로그래머스 문자열 내 p와 y의 개수 파이썬 풀이와 해설


by Kitle · 2020. 07. 19.



원본 출처 프로그래머스 : https://programmers.co.kr/learn/courses/30/lessons/12916

문제 설명

대문자와 소문자가 섞여있는 문자열 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
s = s.upper()
if s.count('P') != s.count('Y'):
answer = False
return answer

해설
개수를 비교할 때 대문자와 소문자는 구별하지 않는 다는 조건이 이 문제의 키 포인트로 볼 수 있습니다.
파이썬의 문자열에서 특정 문자를 세어주는 count()를 이용해 쉽게 풀 수 있습니다.
단 여기서 count는 대/소문자를 구별해줍니다.
따라서 소문자 count 를 모두 세고 대문자 count를 모두 세는 방식으로도 풀 수 있습니다.
그리고 그 부분을 비교해줘도 되겠지만, 대소문자가 상관 없다고 하니, 입력받은 숫자를 모두 대문자로 변환하거나 모두 소문자로 선 변환 후 처리해주는 것이 조건문 작성에 편리함이 생길듯 하네요.
upper()를 이용해 모두 대문자로 변환해줍니다. 물론 소문자로 변환해도 상관은 없습니다.
디폴트로는 True 를 넣어둡니다. 이부부는 문자열이 하나도 없을때 디폴트로 출력되게 하기 위함입니다.
자 그리고 카운트를 통해 값을 비교해봅니다. 모두 대문자 변환하였으므로 대문자를 카운트하여 비교하면 됩니다.
비교한 값이 다른 경우는 False로 변경해줍니다. 
크게 어려움이 없죠?
더 간결하게 사용하기 위해서는 s.upper.count('P') 처럼 써주는 방법도 있겠습니다.