JAVA 백준 2108번 통계학

문제

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

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

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

입력

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

5
1
3
8
-2
2

1
4000

3
0
0
-1

출력

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

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

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

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

2
2
1
10

4000
4000
4000
0

0
0
0
1

알고리즘 분류

구현(implementation), 수학(math), 정렬(sorting)

소스코드

import java.io.*;

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[] list = new int[8001];
        int n = 0;
        int total = 0;
        for (int i = 0; i < N; i++) {
            n = Integer.parseInt(br.readLine());
            list[n + 4000]++;
            total += n;
        }
        // 1
        bw.write(Math.round((float) total / N) + "\n");
        // 2
        int center = 0;
        for (int i = 0; i < list.length; i++) {
            if (list[i] > 0) {
                center += list[i];
                if (center >= (N / 2) + 1) {
                    bw.write((i - 4000) + "\n");
                    break;
                }
            }
        }
        // 3
        int[] freq = {0, 0, 0};
        for (int i = 0, j = 0; i < list.length; i++) {
            if (list[i] != 0) {
                j++;
                if (freq[0] < list[i]) {
                    freq[0] = list[i];
                    freq[2] = i;
                } else {
                    if (freq[0] == list[i] && freq[1] < list[i]) {
                        freq[1] = list[i];
                        freq[2] = i;
                    }
                }
                if (j == N) break;
            }
        }
        bw.write(freq[2] - 4000 + "\n");
        // 4
        int[] minmax = {0, 0};
        for (int i = 0; i < list.length; i++) {
            if (list[i] != 0) {
                minmax[0] = i;
                break;
            }
        }
        for (int i = list.length - 1; i >= 0; i--) {
            if (list[i] != 0) {
                minmax[1] = i;
                break;
            }
        }
        bw.write((minmax[1] - minmax[0]) + "\n");
        bw.flush();
    }
}