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

문자열 내 p와 y의 개수

연습 문제

level 1

 

python3

import re

def solution(s):
    pattern_p = re.compile('[pP]{1}')
    result_p = pattern_p.findall(s)
    pattern_y = re.compile('[yY]{1}')
    result_y = pattern_y.findall(s)

    return len(result_p) == len(result_y)

정규표현식을 이용한 풀이.

 

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

혹은 그냥 lower case로 변경한 뒤, count 함수를 이용해 개수를 세어도 된다.