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

개발자가 되고 싶어요 ㅣ 2023. 3. 3. 06:05

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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));
        StringBuilder sb = new StringBuilder();
 
        // 테스트 갯수 입력받기.
        int n = Integer.parseInt(br.readLine());
 
        // 리스트 생성
        int [] numList = new int [n];
 
        // 리스트에 입력받은 수 넣기
        for (int i = 0; i < n; i++) {
            numList[i] = Integer.parseInt(br.readLine());
        }
 
        // 오름차순으로 정렬
        Arrays.sort(numList);
 
        // 산술평균
        long sum = 0;
        for (int num :
                numList) {
            sum+=num;
        }
        if(sum!=0) {
            sb.append(Math.round((double)sum/n) + "\n");
        } else {
            sb.append(0 + "\n");
        }
 
        // 중앙값
        sb.append(numList[n/2+ "\n");
 
        // 최빈값 ( 여러개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력 )
        int [] modList = new int [n];
 
        for (int i = 0; i < n; i++) {
            modList[i]=1;
        }
 
        for (int i = 0; i < n-1; i++) {
            if (numList[i]==numList[i+1]) {
                modList[i+1]=modList[i]+1;
            }
        }
 
        int maxMod = 0;
        for (int i = 0; i <n ; i++) {
            if (maxMod<modList[i]) {
                maxMod = modList[i];
            }
        }
 
        ArrayList<Integer> resultList = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            if (modList[i] == maxMod) {
                resultList.add(numList[i]);
            }
        }
 
        if (resultList.size()>=2) {
            sb.append(resultList.get(1+ "\n");
        } else {
            sb.append(resultList.get(0+ "\n");
        }
 
        // 범위
        sb.append(numList[n-1]-numList[0]);
 
        System.out.println(sb);
    }
}
cs

 


◎ 기록

 

이번 문제는 최빈값을 구할때 조건(최빈값이 여러개일 경우 최빈값 중 두번째로 작은 값을 출력한다)을 처리하는게 중요했다. 필자는 값을 담을 타입을 배열로 선택했기 때문에 문제 풀이에서 인덱스를 최대한 활용했다. 우선 값이 존재하면 빈도는 무조건 1 이상이므로 전부 1로 설정해준다. 그 후 정렬된 배열을 이용하여 다음 인덱스의 값이 현재 인덱스의 값과 같다면 빈도수를 1씩 늘려준다. 각 빈도수를 모두 구했다면 그 중 최대 빈도수를 저장해준다. 그리고 나서 빈도수를 저장해둔 배열을 순회하며 최대 빈도수와 같은 빈도수를 갖는 값을 어레이리스트에 저장한다. 그리고 나서 최대빈도수를 갖는 값의 개수에 따라 조건을 달아 출력해준다.