Algorithm/JAVA - BOJ

BOJ/백준 - 2108 통계학 JAVA

ㅇㅇ잉 2021. 9. 26. 01:54

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

 

2108번: 통계학

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

www.acmicpc.net

 

 

뭔가 더 깔끔하게 풀 수 있는 방법이 있을 것 같은데,,,,,,

아무튼 여기서 가장 신경써야할 부분은 최빈값인데, '두 번째' 작은 값을 잘 확인해줘야 한다.

설명은 주석~!

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
 
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[] cnt = new int[8001];
 
        int sum = 0//평균 구하기 위해
        int mode = 0//최빈값
 
        for(int i=0;i<N;i++){
            int tmp = Integer.parseInt(br.readLine());
            arr[i]=tmp;
            cnt[tmp+4000]++;
            sum+=tmp;
        }
 
        Arrays.sort(arr);
 
 
        //'두 번째' 작은 값인지 확인하기 위해
        boolean flag = false;
        int max = Integer.MIN_VALUE;
        for(int i=0;i<8001;i++){
            if(cnt[i]==0continue;
 
            //첫번째 작은 값
            if(cnt[i]>max){
                max=cnt[i];
                mode=i-4000;
                flag=true;
            }
            //두번째 작은 값
            else if(flag && max==cnt[i]) {
                mode = i - 4000;
                flag=false;
            }
        }
 
        System.out.println((int) Math.round((double) sum / N));
        System.out.println(arr[N/2]);
        System.out.println(mode);
        System.out.println(arr[N-1]-arr[0]);
    }
}
cs