출처 | 통계학 [백준 2108]
수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.
N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.
첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.
첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.
둘째 줄에는 중앙값을 출력한다.
셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.
넷째 줄에는 범위를 출력한다.
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 scan = new BufferedReader(new InputStreamReader(System.in));
int N = Integer.parseInt(scan.readLine());
ArrayList<Integer> arr = new ArrayList<>();
for (int i = 0; i < N; i++) {
arr.add(Integer.parseInt(scan.readLine()));
}
StringBuilder sb = new StringBuilder();
// 산술평균
long sum = 0;
for (int i = 0; i < N; i++) {
sum += arr.get(i);
}
sb.append(Math.round((double)(sum)/N) + "\n");
Collections.sort(arr);
// 중앙값
sb.append(arr.get(N/2) + "\n");
// 최빈값
int count = 1;
int max = 0;
ArrayList<Integer> mostFrequency = new ArrayList<>();
for (int i = 0; i < N; i++) {
if (i == N - 1) {
if (max == count) {
mostFrequency.add(arr.get(i));
}
else if (max < count) {
mostFrequency.clear();
mostFrequency.add(arr.get(i));
}
break;
}
if (arr.get(i + 1).equals(arr.get(i))) {
count++;
}
else if (!arr.get(i + 1).equals(arr.get(i))) {
if (max == count) {
mostFrequency.add(arr.get(i));
count = 1;
}
else if (max < count) {
mostFrequency.clear();
mostFrequency.add(arr.get(i));
max = count;
count = 1;
}
else {
count = 1;
}
}
}
Collections.sort(mostFrequency);
if (mostFrequency.size() > 2) {
sb.append(mostFrequency.get(1) + "\n");
}
else if (mostFrequency.size() == 1){
sb.append(mostFrequency.get(0) + "\n");
}
else if (mostFrequency.size() == 2) {
sb.append(mostFrequency.get(1) + "\n");
}
// 범위
sb.append(arr.get(N - 1) - arr.get(0));
System.out.print(sb.toString());
}
}