99클럽 코테 스터디 3일차 TIL + 프로그래머스/1/12916. 문자열 내 p와 y의 개수

문제 설명

대문자와 소문자가 섞여있는 문자열 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

입출력 예 설명

입출력 예 #1
'p'의 개수 2개, 'y'의 개수 2개로 같으므로 true를 return 합니다.

입출력 예 #2
'p'의 개수 1개, 'y'의 개수 2개로 다르므로 false를 return 합니다.

※ 공지 - 2021년 8월 23일 테스트케이스가 추가되었습니다.

 

 

<내코드>

def solution(s):
    
    s = s.lower()
    # if 'p' not in s and 'y' not in s :
    #     return False
    
    p_num = 0
    y_num = 0
    
    for char in s:
        if 'p' in char :
            p_num += 1
        elif 'y' is char :
            y_num += 1
    
    if p_num == y_num :
        return True
    else :
        return False

처음에는 p,y 가 모두 없을 때 즉 0 일때에 대한 조건을 만들었음

그런데 사실 두개의 수가 같을 때 즉 두개다 0일 때도 마찬가지로 0임 그래서 특별히 더 조건을 설정할 필요가 없음

 

<모범사례>

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

count() 라는 메소드로 한번에 문자열에 대해서 해당 문자열 내 특정문자의 갯수에 대해서 찾기 가능함

count()를 조사해보니 좀 더 다양하게 쓸수있는것 같다

 

<추가학습>

count

 

dictionary 와 set 자료형 제외 iterable 자료형에 대해서 (즉 문자열,튜플,리스트,집합)

list.count('a') 하면 list 안에 a 문자의 갯수가 리턴이 됨