슬기로운 코테 생활
[프로그래머스] 문자열 내 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