programmers/level1

[level1] - 문자열 내 p와 y의 개수

태기의삶 2020. 2. 25. 23:42

 

 

프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr

 

[풀이]

<핵심>

'p'의 개수와 'y'의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요.

단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다.

 

예를 들어, 

"pPoooyY"이라는문자열을 입력 받았을 때,

여기서 p의 대소문자의 개수를 구해서 P라는 변수에 담고,

y의 대소문자의 개수를 구해서 Y라는 변수에 담았다.

P와 Y의 개수가 같으면 true를 return 하고,

P와 Y의 개수가 틀리면 false를 return 하는 로직을 생각해보았다.

 

 

[code]

def solution(s):
    answer = True
    P = s.count('p');
    P += s.count('P');
    Y = s.count('y');
    Y += s.count('Y');
    if P == Y:
        return answer
    else:
        answer = False
        return answer