[Programmers] 최빈값 구하기(자바)

문제

 

최빈값은 주어진 값 중에서 가장 자주 나오는 값을 의미합니다. 정수 배열 array가 매개변수로 주어질 때, 최빈값을 return 하도록 solution 함수를 완성해보세요. 최빈값이 여러 개면 -1을 return 합니다.

 

 

제한사항

 

   -   0 < array의 길이 < 100

   -   0 ≤ array의 원소 < 1000

 

 

---

입출력 예

array	result
[1, 2, 3, 3, 3, 4]	3
[1, 1, 2, 2]	-1
[1]	1

 

---

입출력 예 설명

  입출력 예 #1

• [1, 2, 3, 3, 3, 4]에서 1은 1개 2는 1개 3은 3개 4는 1개로 최빈값은 3입니다.

  입출력 예 #2

• [1, 1, 2, 2]에서 1은 2개 2는 2개로 최빈값이 1, 2입니다. 최빈값이 여러 개이므로 -1을 return 합니다.

  입출력 예 #3

• [1]에는 1만 있으므로 최빈값은 1입니다.

 

 

---

 

풀이

💡 접근 방법

  1. 배열에서의 최대값을 구한다. (max)

  2. 최대값만큼의 길이를 갖는 배열을 할당한다. (cnt[ ])

      (* 최대값의 index에 최대값이 들어오도록 +1을 해준다)

  3. for문으로 주어진 배열의 각 값과 일치하는 부분을 증가시킨다. (발생 횟수를 카운트 함)

  4. 배열(cnt[ ])에서 가장 큰 값의 index를 구한다.

  5. 구하는 과정에서 가장 큰 값과 동일한 값이 존재한다면 -1을 return 한다.

 

 

처음 에러 코드

import java.util.Arrays;

class Solution {
    public int solution(int[] array) {
        int answer = 0;
        int max = 0;
        
        //배열에서의 최대값 구하기
        for(int i=0; i<array.length; i++){
            if(max < array[i]){
                max = array[i];
            }
        }
        
        int cnt[] = new int[max+1];
        for(int i=0; i<array.length; i++){
            cnt[array[i]]++;
        }
        Arrays.sort(cnt);
        
        // !문제발생지점
        if(cnt[cnt.length-1] == cnt[cnt.length-2]){
            answer = -1;
        } else{
            answer = cnt[cnt.length-1];
        }
        
        return answer;
    }
}

오류가 발생한 코드이다.

 

cnt[ ]을 가장 많이 count된 순서대로 오름차순 정렬을 하여, 마지막 값과 직전 값이 같다면 최빈값이 여러 개 이므로 -1을 출력하도록 했다.

 

!문제:  주어진 테스트케이스 3개는 통과했으나, 제출 시 16개 중 6개의 테스트만 통과하였다.

!원인:  최빈값이 여러 개 일 때 -1을 출력하도록 한 코드에서

          배열에 값이 1개만 들어있을 경우 .length-2를 했을 때 해당하는 index가 존재하지 않게 되어

          오류가 발생할 수 밖에 없었다.

 

 

 

최종 코드

class Solution {
    public int solution(int[] array) {
        int answer = 0;
        int max = 0;
        int max_cnt = 0;
        
        //배열에서의 최대값 구하기
        for(int i=0; i<array.length; i++){
            if(max < array[i]){
                max = array[i];
            }
        }
        
        int cnt[] = new int[max+1];
        for(int i=0; i<array.length; i++){
            cnt[array[i]]++;
        }
        
        for(int i=0; i<cnt.length; i++){
            if(max_cnt < cnt[i]) {
                max_cnt = cnt[i];
                answer = i;
            } else if(max_cnt == cnt[i]) {
                answer = -1;
            }
        }
        return answer;
    }
}

오름차순 정렬 대신 cnt[ ]에서 가장 큰 값을 찾는다. 값을 찾는 도중 동일한 값이 존재한다면 -1을 return 하도록 수정했다.