[코딩테스트 입문] 문자열 내 p와 y의 개수(Python3)

[코딩테스트 입문] 문자열 내 p와 y의 개수(Python3)

---

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

프로그래머스

 

count 함수 써서 했다.

그래서 대문자 소문자 구분해서 count 해줬는데 다른사람 코드를 보니까 lower을 통해 소문자로 만들어줬더라!

 

def solution(s):
    num_p = s.count('p')
    num_P = s.count('P')
    num_y = s.count('y')
    num_Y = s.count('Y')
    
    p = num_p + num_P
    y = num_y + num_Y
    
    if p == y or p+y==0:
        return True
    else:
        return False

 

lower() 함수 이용해서 count 하는 방법!

def solution(s):
    if s.lower().count('p') == s.lower().count('y') or s.lower().count('p')+s.lower().count('y')==0:
        return True
    else:
        return False