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

말하는 감자·2022년 6월 27일

Programmers Level 1

목록 보기

24/66

프로그래머스 Level 1

🔒 문자열 내 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 합니다.

🗝️ Python 작성 코드

def solution(s):
    answer = True
    
    s = s.lower()
    
    p = s.count('p')
    y = s.count('y')
    
    if(p != y):
        answer = False

    return answer

파이썬이면 쉽게 풀 수 있겠는 걸? 싶어서 파이썬으로 먼저 풀어봤다.

대소문자를 구분하지 않는다고 했으니 string.lower()를 이용해 모두 소문자로 변경하여 대소문자를 통일하고
string.count()를 통해 문자열에 해당 문자가 몇개 들어있는지 가져온다.

그리고 p의 개수와 y의 개수가 안맞는다면 answer를 False로 변경하여 리턴한다.

🗝️ JAVA 작성 코드

class Solution {
    boolean solution(String s) {
        boolean answer = true;
        int p = 0;
        int y = 0;

        s = s.toLowerCase();
        
        for(int i=0; i<s.length(); i++) {
            if(s.charAt(i) == 'p') p++;
            else if(s.charAt(i) == 'y') y++;
        }
        
        if(p != y) answer = false;

        return answer;
    }
}

파이썬 코드와 사실상 동일하다.
s를 String.toLowerCase()을 이용하여 대소문자를 소문자로 통일시킨 후 s의 길이만큼 반복문을 돌려 p나 y가 일치할 때 마다 각 개수를 카운트한다.

그리고 p의 개수와 y의 개수가 안맞는다면 answer를 false로 변경하여 리턴한다.

🔓 JAVA 다른 사람의 코드

class Solution {
    boolean solution(String s) {
        s = s.toLowerCase();
        int count = 0;

        for (int i = 0; i < s.length(); i++) {

            if (s.charAt(i) == 'p')
                count++;
            else if (s.charAt(i) == 'y')
                count--;
        }

        return count == 0;
    }
}