[백준/Java] 2108번 :: 통계학

| 문제

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

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

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

 

| 입력

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

 

| 출력

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

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

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

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

 

| 풀이

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

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

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

        int[] arr = new int[N];
        int sum = 0;    // 평균을 구하기 위한 sum
        int[] num = new int[8001];  // 최빈값 구하기 위한 배열

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

        // 산술평균 구하기
        int avg = (int) Math.round((double)sum/N);

        // 중앙값 구하기
        Arrays.sort(arr);
        int center = arr[N/2];

        // 최빈값 구하기
        int max = 0;
        int count = 0;
        boolean flag = false;

        for(int i=0; i<num.length; i++)
        {
            if(num[i] > max){
                count = i-4000;
                max = num[i];
                flag = true;
            }
            else if(num[i]==max && flag) {
                count = i-4000;
                flag = false;
            }
        }

        // 범위 구하기
        int range = arr[N-1] - arr[0];

        sb.append(avg).append("\n");
        sb.append(center).append("\n");
        sb.append(count).append("\n");
        sb.append(range);

        System.out.println(sb);

    }
}

 

| 정리

최빈값이 여러개라면 그 중 두번째로 작은 값을 구하는 것이 신경써야 할 문제인 것 같다.

flag 로 첫번째 나온 최빈값인지 두번째인지 판별하는 법으로 풀이했다.