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

내 코드

class Solution {
    boolean solution(String s) {
        boolean answer = true;

        int[] cnt = new int[2];
        
        s = s.toLowerCase();
        
        for(int i=0 ; i < s.length() ; i++) {
            if(String.valueOf(s.charAt(i)).equals("p"))      cnt[0]++;
            else if(String.valueOf(s.charAt(i)).equals("y")) cnt[1]++;
        }
        
        if(cnt[0] == cnt[1] && cnt[0]!= 0 && cnt[1] != 0) answer = true;
        else if(cnt[0] != cnt[1])                         answer = false;
        else if(cnt[0] == 0 && cnt[1] == 0)               answer = true;
        
        // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
        System.out.println("Hello Java");

        return answer;
    }
}

String.valueOf(s.charAt(i)).equals("p") 대신

s.charAt(i) == 'p' 이렇게 해줘도 됌...

 

다른 분들 코드 보니까 아예 변수 하나만 가져가신 분도 있었다

프로그래머스

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;
    }
}

그리고 내가 마지막 줄에 휘갈겨 놓은 if문을 이렇게 return 에서 끝내도 되었던 것이었다ㅎ