2108번: 통계학

문제 출처 : https://www.acmicpc.net/problem/2108

 

언어 : Kotlin

 

문제 설명 :

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

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

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

• 첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.
• 첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.
• 둘째 줄에는 중앙값을 출력한다.
• 셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.
• 넷째 줄에는 범위를 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
5 1 3 8 -2 2	2 2 1 10
1 4000	4000 4000 4000 0
5 -1 -2 -3 -1 -2	-2 -2 -1 2
3 0 0 -1	0 0 0 1

풀이 : 

import java.io.*
import kotlin.math.roundToInt

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val n = br.readLine().toInt()
    val arr = IntArray(n) { br.readLine().toInt() }

    val freqMap = arr.toMutableList().groupingBy { it }.eachCount()
    val mapKey = freqMap.filter { it.value == freqMap.maxOf { it.value } }.keys.sorted()

    bw.appendLine("${arr.average().roundToInt()}")
        .appendLine("${arr.sortedArray()[arr.size / 2]}")
        .appendLine("${if (mapKey.size == 1) mapKey.first() else mapKey[1]}")
        .appendLine("${arr.max() - arr.min()}")
    bw.flush()
    bw.close()
}

최빈값 쪽에서 양수 쪽은 괜찮았으나, 음수가 나올 때 조금 많이 헤매었다...

다른 사람들의 풀이를 참고한 결과, MutableList의 eachCount라는 유용한 함수를 통해 코드를 간략하게 풀어낼 수 있었다.

 

참고한 풀이 : https://velog.io/@kldaji/%EB%B0%B1%EC%A4%80-2108%EB%B2%88-%ED%86%B5%EA%B3%84%ED%95%99

[백준] 2108번: 통계학