[BaekJoon] 2108 통계학 JAVA

🔗 백준 2108 통계학 https://www.acmicpc.net/problem/2108

문제

수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.

1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
4. 범위 : N개의 수들 중 최댓값과 최솟값의 차이

N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

---

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.

---

출력

첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.

둘째 줄에는 중앙값을 출력한다.

셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.

넷째 줄에는 범위를 출력한다.

---

풀이

평균값 : 입력받은 모든 수를 더하여 나눈 평균을 round로 소수점을 반올림하여 출력 중앙값 : 입력받은 수가 들어있는 배열을 정렬한 후 n/2에 위치한 값을 출력 범위 : 정렬한 배열에서 가장 큰 수 - 가장 작은 수 를 출력

여기서 문제는 최빈값이다. 여러개가 있다면 최빈값 중 두 번째로 작은 값을 출력해야한다. 여기서 고민하는 동안 시간을 조금 잡아먹었다.

먼저 최빈값을 구하기 위해 수를 입력받을 때마다 그 수를 index로 하여 카운팅할 boolean 배열을 선언해주었다. 그 후로 카운팅을 끝낸 후, boolean 배열에서 가장 큰 수를 찾아 그 인덱스를 list에 저장한다.

만약 list의 길이가 1보다 크다면 1번 인덱스에 저장한 수를, 아니라면 0번째 수를 출력시켜준다.

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n  = Integer.parseInt(br.readLine());
        int[] arr = new int[n];
        List<Integer> list = new ArrayList<>();
        int[] check = new int[8001]; // 절대값 4000, 즉 -4000 ~ 4000까지 총 8001개
        int max = 0;
        int sum = 0;

        for (int i = 0; i < n; i++) {
            arr[i] = Integer.parseInt(br.readLine());
            sum += arr[i];
            check[arr[i] + 4000]++;
        }

        System.out.println((int)Math.round((double)sum/n)); // 평균
        Arrays.sort(arr);
        System.out.println(arr[n/2]); // 중앙값

        for (int i = 0; i < 8001; i++) max = Math.max(max, check[i]);
        for (int i = 0; i < 8001; i++) if(check[i] == max) list.add(i - 4000);

        System.out.println(list.size() > 1 ? list.get(1) : list.get(0)); // 최빈값
        System.out.println(arr[arr.length-1] - arr[0]); // 범위
    }
}