2108 JAVA 통계학

2108번: 통계학

🍀 문제

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

산술평균 : N개의 수들의 합을 N으로 나눈 값
중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
최빈값 : N개의 수들 중 가장 많이 나타나는 값
범위 : N개의 수들 중 최댓값과 최솟값의 차이
N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

🍀 입력

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

🍀 출력

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

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

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

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

🍀 예제

예제 입력 1

5
1
3
8
-2
2

예제 출력 1

2
2
1
10

예제 입력 2

1
4000

예제 출력 2

4000
4000
4000
0

예제 입력 3

5
-1
-2
-3
-1
-2

예제 출력 3

-2
-2
-1
2

예제 입력 4

3
0
0
-1

예제 출력 4

0
0
0
1

🍀 코드

나머지는 다 쉽게 해결이 가능하지만, 최빈값의 경우에는 값을 저장하고 비교하는 과정이 필요하다. key, value 사용을 위해 HashMap을 이용했고, 이를 통해 입력된 각 숫자의 빈도를 저장한다.
이후, 최빈도를 찾아서 최빈도를 갖고있는 숫자들 중에 두 번째로 작은 숫자(Key)를 선택한다. 만약, 최빈도를 갖고 있는 숫자가 1개라면 그 숫자를 출력한다.

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());
        
        ArrayList<Integer> ar = new ArrayList<>();
        HashMap<Integer, Integer> map = new HashMap<>();
        
        double sum = 0;
        for(int i = 0; i < n; i++){
            int num = Integer.parseInt(br.readLine());
            ar.add(num);
            sum += num;
            // 최빈값을 위해 저장
            if(map.containsKey(num)) map.replace(num, map.get(num) + 1);
            else map.put(num, 1);
        }

        // 최빈값 구하기
        int max = 0;
        for (int value : map.values()){
            max = Math.max(max, value);
        }
        List<Integer> li = new ArrayList<>();
        for (int key : map.keySet()){
            if(map.get(key) == max){
                li.add(key);
            }
        }
        Collections.sort(li);


        // 정렬
        Collections.sort(ar);
        // 산술 평균
        System.out.println(Math.round(sum/n));
        // 중앙값
        System.out.println(ar.get(((int)ar.size()/2)));
        // 최빈값
        System.out.println(li.size() == 1 ? li.get(0) : li.get(1));
        // 범위
        System.out.println(Math.abs(ar.get(ar.size()-1)-ar.get(0)));
    }
}