[프로그래머스] 문자열 내의 p와 y의 갯수

def solution(s):
    cnt_p = 0
    cnt_y = 0
    s = s.lower()
    answer = False
    
    for character in s:
        if character == "p":
            cnt_p += 1
        elif character == "y":
            cnt_y += 1
        else:
            pass
        
    if cnt_p == cnt_y:
        answer = True
        
    return answer

 

count가 가능한지 모르겠던 나머지 주먹구구식 코드를 짜버렸다.

다른사람 코드를 확인해보니 count가 가능하더라.

 

이래서 사람이 기본적으로 사용 가능한 함수들을 다 기억해둬야 야무지게 코드를 짜는 것 같다.

 

코드를 차차 줄여나가보면

def solution(s):
    s = s.lower()
    answer = False
    
    if s.count("p") == s.count("y"):
        answer = True
        
    return answer

 

먼저 소문자로 바꿔주고 "p"와 "y"의 갯수를 확인을 해주었을 때 같으면 answer의 값을 true로 반환

def solution(s):
    answer = False
    
    if s.lower().count("p") == s.lower().count("y"):
        answer = True
        
    return answer

 

lower case로 바꿔주는건 if문 안에서 돌려줘도 되고

def solution(s):
    return s.lower().count("p") == s.lower().count("y")

 

최종적으로 이렇게 줄여준다면? 결국은 True, False를 반환하게 된다.

 

오늘의 배움점

문자열 count가 가능하다!