[JAVA] 최빈값 구하기

문제 설명

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

---

제한사항

• 0 < array의 길이 < 100
• 0 ≤ array의 원소 < 1000

---

입출력 예arrayresult

[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입니다.

---

import java.util.Arrays;

class Solution {
    public int solution(int[] array) {
        int answer = 0;
        int max = 0; 
        
        Arrays.sort(array); // 배열 오름차순 정렬
        max = array[array.length-1]; // 가장 큰 수를 max로 설정
            
        // 각 수의 count를 위한 배열
        int count[] = new int[max + 1]; 
        for(int n: array) {  
            count[n]++; // 각 수의 idx에 count
        }
        max = count[0];
        // 가장 큰 count 값을 max로 설정
        for(int i=1; i<count.length;i++) {
            if(count[i]>max) {
                max = count[i];
                answer = i; // max 값이 바뀔 때 i는 최빈값
            }else if(max == count[i]) { // 최빈값이 여러개일 때 -1
                answer = -1;
            }
        }
        return answer;
    }
}

수많은 테스트 케이스때문에 실패.. 화난다.. 코드를 참고를 했는데.. 계속 봐야겠다.. 

테스트케이스에 실패한 코드..

class Solution {
    public int solution(int[] array) {
        int answer = 0;
        int arr_max = 0; 

        for (int i = 0; i < array.length; i++) {
            if (array[i] > arr_max) {
                arr_max = array[i];
            }
        }
        int[] compareArr = new int[arr_max+1];

        // array의 개수가 1 이상일 때만 
        if(array.length > 1) { 
            int max = 0;    // 최빈값
            int count = 0;  // 최빈값의 개수

            // array에 있는 값들을 비교하기 위해 compareArr에 넣기
            for(int n: array) { 
                compareArr[n]++;
            }
            // 최빈값을 max에 넣기
            for(int i=0;i<compareArr.length;i++) {
                if(compareArr[i]>max) {
                    max = compareArr[i];
                    answer = i;
                }
            }
            // 최빈값의 개수 구하기(최빈값의 개수가 같으면 count ++ )
            for(int n : compareArr){
                if(n == max) count ++;
                if(count > 1) answer = -1;
            }

        }else answer = 1;

        return answer;
    }
}

 

참고

기초알고리즘 최빈값