[연습문제] 문자열 내 p와 y의 개수

https://school.programmers.co.kr/learn/courses/30/lessons/12916

프로그래머스

---

문제 설명
대문자와 소문자가 섞여있는 문자열 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 합니다.

---

대문자와 소문자가 섞여있는 문자열 s가 주어질 때, 이 s 안에 포함되어 있는 'p', 'y' 문자의 수가 같은지 다른지 확인해야 하는 문제입니다.

 

이 문제를 자바로 푼다면 toLowerCase/toUpperCase를 사용해 대문자/소문자로 통일한 뒤, 반복문이나 chars를 사용해 값을 하나씩 비교하면서 개수를 셀 것 같습니다.

파이썬도 upper/lower을 사용해 전부 대문자/소문자로 변환한 뒤, count를 사용해 총 몇 개인지 구해서 비교할 것 같네요.

 

정답 코드를 작성해보겠습니다.

 

1. 자바

class Solution {
    boolean solution(String s) {
        int pCount = (int) s.toLowerCase().chars().filter(c -> c == 'p').count();
        int yCount = (int) s.toLowerCase().chars().filter(c -> c == 'y').count();

        if (pCount == yCount)
            return true;
        return false;
    }
}

 

2. 파이썬

def solution(s):  
    pCount = s.lower().count('p')
    yCount = s.lower().count('y')
    
    if pCount == yCount:
        return True
    return False

 

다른 언어로 풀 때마다 파이썬이 얼마나 간단한지 다시금 실감합니다.