슬기로운 연구생활

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

슬기로운 코테 생활

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

vhrehfdl 2020. 11. 17. 10:59

* 문제

programmers.co.kr/learn/courses/30/lessons/12916

 

코딩테스트 연습 - 문자열 내 p와 y의 개수

대문자와 소문자가 섞여있는 문자열 s가 주어집니다. s에 'p'의 개수와 'y'의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요. 'p', 'y' 모두 하나도 없는 경우는 항상 True를

programmers.co.kr

 

* 문제 풀이 

- 첫째, 반복문을 사용해서 p와 y의 개수를 카운트한다.

- 둘째, 조건문을 사용해서 p와 y가 같은 경우 True와 Fasle를 분류한다.

 

* 생각

- 다른 사람의 풀이를 보니 count 함수를 사용해서 풀었다.

함수를 더 많이 알자.

 

* 코드 

def solution(s):
    answer = True
    
    s = s.lower()
    
    p_cnt, y_cnt = 0, 0
    for i in range(0, len(s)):
        if s[i] == "p":
            p_cnt += 1
            
        elif s[i] == "y":
            y_cnt += 1
    
    if p_cnt == y_cnt:
        answer = True
    else:
        answer = False
    
    return answer

 

Comments