백준 2108(JAVA)

문제

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

2108번: 통계학

풀이

• 산술평균

1. 더해서 나누면 되는데, -1.8인 경우 반올림하면 -1이아닌 -2가 되어야함. 이 부분을 처리하기 위해 나는 나눈 값 절대값으로 바꾼 뒤, Math.round를 이용
2. 오랜만에 자바를 해서 그런지 나눌때 auto casting되는걸 까먹었당. int / int => int가 된다. 그래서 피연산자 둘중 하나를 명시적으로 double로 casting 후 나눠야한다.

• 중앙값

1. ArrayList를 이용해 Collections sort 로 정렬후 중앙값 추출(N: 무조건 홀수)

• 최빈값

1. Map을 이용해서 풀었다. Map의 key는 입력받은 수이고, value는 해당 key의 빈도 수를 저장함.
2. 문제를 꼼꼼히 읽지않아, 최빈값이 여러개 일 때, 두번째로 큰 값을 출력하는걸 몰랐다.

• 범위

1. 위의 정렬된 ArrayList를 재사용함

코드

package baekjoon.구현;

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

public class Main2108 {
    static int N;
    static List<Integer> list = new ArrayList<>();
    static Map<Integer, Integer> map = new HashMap<>();

    static int maxCount = 0;
    static int maxKey = -10000;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());

        for(int i = 0; i < N; i++){
            int value = Integer.parseInt(br.readLine());
            list.add(value);
            if(map.containsKey(value)){
                map.put(value, map.get(value) + 1);
            }else{
                map.put(value, 1);
            }
        }

        Collections.sort(list);

        findMaxCountAndMaxKey();
        List<Integer> maxLists = getMaxCount();

        int sum = list.stream().mapToInt(Integer::intValue).sum();

        if(sum < 0){
            double result = (double)sum / N;
            double value = Math.abs(result);
            System.out.println(Math.round(value) * -1);
        }else{
            System.out.println(Math.round((double)sum / N));
        }
        System.out.println(list.get(N / 2));
        if(maxLists.size() == 1){
            System.out.println(maxLists.get(0));
        }else{
            System.out.println(maxLists.get(1));
        }
        System.out.println(list.get(N - 1) - list.get(0));
    }
    static void findMaxCountAndMaxKey(){
        for(Map.Entry<Integer, Integer> entry: map.entrySet()){
            int value = entry.getValue();
            int key = entry.getKey();
            if(maxCount < value){
                maxCount = value;
                maxKey = key;
            }
        }
    }

    static List<Integer> getMaxCount(){
        List<Integer> maxLists = new ArrayList<>();

        for(Map.Entry<Integer, Integer> entry: map.entrySet()){
            Integer value = entry.getValue();
            if(value == maxCount){
                maxLists.add(entry.getKey());
            }
        }

        Collections.sort(maxLists);

        return maxLists;
    }
}