[BOJ][Java] 2108번 - 통계학

문제

• 백준 2108번: 통계학
• 산술평균, 중앙값, 최빈값, 범위를 구하는 문제

풀이 설명

• 최빈값을 구하는 조건이 까다로워서 두가지 방법으로 풀었다.
• 첫번째 방법은 필요 이상으로 복잡해서 시간제한에도 아슬아슬하게 맞았다.
  1) 최빈값 구할 때 HqshMap, LinkedList, Comparator 사용
  2) 최빈값 구할 때 ArrayList 사용 (참고: https://qlyh8.tistory.com/139)

성공 코드

1) 최빈값 구할 때 HqshMap, LinkedList, Comparator 사용

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader((System.in))); 
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter((System.out)));
        int n = Integer.parseInt(br.readLine());
        int[] nums = new int[n];

        for (int i = 0; i < n; i++) {
            nums[i] = Integer.parseInt(br.readLine());
        }
        br.close();
        Arrays.sort(nums);

        bw.write(mean(nums) + "\n");
        bw.write(median(nums) + "\n");
        bw.write(mode(nums, n) + "\n");
        bw.write(diff(nums) + "\n");
        bw.close();
    }

    public static int mean(int[] nums) {
        double sum = 0.0;
        for (int num : nums)
            sum += num;

        return (int) Math.round(sum / nums.length);
    }

    public static int median(int[] nums) {
        return nums[ nums.length / 2 ];
    }

    public static int mode(int[] nums, int n) {
        boolean visited[] = new boolean[n];
        Arrays.fill(visited, false);

        // 입력숫자, 숫자의 개수
        HashMap<Integer, Integer> modes = new HashMap<>();

        if (n == 1) return nums[0];
        else {
            for (int i = 0; i < n; i++) {
                if (visited[i] == true)
                    continue;

                int cnt = 1;
                for (int j = i+1; j < n; j++) {
                    if (nums[i] == nums[j]) {
                        visited[j] = true;
                        cnt++;
                    }
                }
                modes.put(nums[i], cnt);
            }
            List<Map.Entry<Integer, Integer>> answer = new LinkedList<>(modes.entrySet());

            Collections.sort(answer, new Comparator<Map.Entry<Integer, Integer>>() {
                @Override
                //key: 입력한 숫자 값들
                //value: key 값들의 개수
                public int compare(Entry<Integer, Integer> arg0, Entry<Integer, Integer> arg1) {
                    if (arg0.getValue() == arg1.getValue())
                        return arg0.getKey() - arg1.getKey();
                    return arg1.getValue() - arg0.getValue();
                }
            });

            if (answer.get(0).getValue() == answer.get(1).getValue())
                return answer.get(1).getKey();
            else
                return answer.get(0).getKey();
        }
    }

    public static int diff(int[] nums) {
        // 최댓값, 최솟값 찾기
        int max = -4001;
        int min = 4001;

        for (int i = 0; i < nums.length; i++) {
            if (max < nums[i]) max = nums[i];
            if (min > nums[i]) min = nums[i];
        }
        return max - min;
    }
}

2) 최빈값 구할 때 ArrayList 사용

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader((System.in))); 
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter((System.out)));
        int n = Integer.parseInt(br.readLine());
        int[] nums = new int[n];
         int visited[] = new int[8001];
       double sum = 0.0;

        for (int i = 0; i < n; i++) {
            nums[i] = Integer.parseInt(br.readLine());
            visited[nums[i] + 4000]++;
            sum += nums[i];                                                                                                                                                                                                                  
        }
        br.close();
        Arrays.sort(nums);

        bw.write((int) Math.round(sum/n) + "\n");
        bw.write(nums[n/2] + "\n");
        if (n == 1) 
            bw.write(nums[0] + "\n");
        else 
            bw.write(mode(visited) + "\n");
        bw.write(nums[n-1] - nums[0] + "\n");
        bw.close();
    }

    public static int mode(int[] visited) {
        ArrayList<Integer> modes = new ArrayList<>();        
        int maxMode = 0;

        for (int i = 0; i <= 8000; i++) {
            if (visited[maxMode] < visited[i]) {
                maxMode = i;
                modes.clear();
            }
            else if (visited[i] != 0 && visited[i] == visited[maxMode]) {
                modes.add(i - 4000);
            }
        }
        if (modes.size() != 0) 
            return modes.get(0);
        else 
            return maxMode - 4000;
    }
}