백준 2108 통계학 java 풀이

문제

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

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

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

입력

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

출력

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

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

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

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

 

나의 풀이

 문제는 이전에 풀었던 카운터정렬 문제와 유사하고 단지 정렬된 배열에 어찌 접근할 것인지 정도만 추가된 문제였다. 주어진 값들을 읽어서 배열에 저장하고, 카운터 정렬을 실행 하였다. 주어지는 값이 -4000 ~ 4000 이기 때문에 배열의 인덱스와 맞추기 위해 모든 값에 4000을 더하며 카운터 정렬을 실행했다. 카운터 정렬 후 가장 빈도가 높은 값을 List에 저장하고, List의 길이를 체크해서 최빈값 중 두번째로 작은 값을 찾았다. 처음 접근법을 생각한 후 어렵지 않게 코드가 작성되어서 다른 사람들의 풀이와 비교해보았지만 대동소이 했고 실행 시간은 상당히 빠른편이었다. 단 모듈화가 되어 있지 않은점은 아쉬운점.

 

코드

import java.util.*;
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 num = Integer.parseInt(br.readLine());
        int[] arr = new int[num];
        
        int sum=0;
        for(int i=0; i<num; i++) {
            arr[i] = Integer.parseInt(br.readLine());
            sum += arr[i];
        }
        
        int[] countArr = new int[8001];
        for(int n : arr) {
            countArr[n+4000]++;
        }
        
        int[] ansArr = new int[num];
        int tempMax = 0, tempNum=0 , index=0;
        for(int i=0; i<countArr.length; i++) {
            tempNum = countArr[i];
            if(tempNum>0) {
                if(tempNum > tempMax) {
                    tempMax = tempNum;
                }
                
                while(tempNum>0) {
                    ansArr[index] = i-4000;
                    tempNum--;
                    index++;
                }
            }
        }
        
        ArrayList<Integer> list = new ArrayList<>();
        int most;
        for(int i=0; i<countArr.length; i++) {
            if(countArr[i]==tempMax) {
                list.add(i-4000);
            }
        }
        
        if(list.size()>1) {
            most = list.get(1);
        } else {
            most = list.get(0);
        }
        
        bw.write((int)Math.round((double)sum/num)+"\n");
        bw.write(ansArr[ansArr.length/2]+"\n");
        bw.write(most+"\n");
        bw.write(ansArr[ansArr.length-1]-ansArr[0]+"\n");
        
        
        
        bw.flush();
    }
}