[백준] 2108번 - 통계학 [Java]

https://www.acmicpc.net/problem/2108

 

[백준] 2108번 - 통계학 [Java]

[백준] 2108번 - 통계학 [Java]

---

1.  아이디어

 

산술평균은 Math.round 메서드로 반올림을 하는 방식으로, 중앙값은 정렬을 활용해서, 최빈값은 카운팅 배열로, 범위는 정렬을 활용해서 구할 수 있다.

---

2. 문제풀이

 

입력 단계에서 수들을 배열로 입력받으면서 합과 카운팅 배열을 작성했다. 이후 배열을 정렬해서 중앙값과 범위를 구하고 합으로 산술평균을 구하고, 카운팅 배열로 최빈값을 찾는 방식으로 구현했다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {

    private static final int MAX = 4000;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        int N = Integer.parseInt(br.readLine());

        int sum = 0;
        int[] arr = new int[N];
        int[] cntArr = new int[MAX + 1 + MAX];
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(br.readLine());
            sum += arr[i];
            cntArr[MAX + arr[i]]++;
        }
        Arrays.sort(arr);

        int max = 0;
        int cnt = 0;
        for (int i = 0; i < cntArr.length; i++) {
            if (cntArr[i] > max) {
                max = cntArr[i];
                cnt = 1;
            } else if (cntArr[i] == max) {
                cnt++;
            }
        }
        cnt = Math.min(cnt, 2);

        int mode = 0;
        for (int i = 0; i < cntArr.length; i++) {
            if (cntArr[i] == max) cnt--;

            if (cnt == 0) {
                mode = i - MAX;
                break;
            }
        }

        sb
                .append(Math.round((double) sum / N)).append("\n")
                .append(arr[N / 2]).append("\n")
                .append(mode).append("\n")
                .append(arr[N - 1] - arr[0]).append("\n");

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기