[백준] 2108번: 통계학

문제

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

풀이

1. 숫자 리스트 입력 받기
2. 평균 계산
3. 중간값 계산
4. Map 을 사용한 빈도수 계산
5. 최댓값 - 최솟값 계산

import kotlin.math.roundToInt

fun getAverage(numberList: MutableList<Int>): Int = numberList.average().roundToInt()

fun getMid(numberList: MutableList<Int>): Int {
    val sortedList = numberList.sorted()
    val mid = numberList.size / 2
    return sortedList[mid]
}

fun getFreq(numberList: MutableList<Int>): Int {
    val freqMap = numberList.groupingBy { it }.eachCount()
    val maxValue = freqMap.maxOf { it.value }
    val maxFreqMap = freqMap.filter { it.value == maxValue }
    val mapKeys = maxFreqMap.keys.sorted()
    return if (mapKeys.size == 1) {
        mapKeys.first()
    } else {
        mapKeys[1]
    }
}

fun getRange(numberList: MutableList<Int>): Int {
    val maxValue = numberList.maxOf { it }
    val minValue = numberList.minOf { it }
    return maxValue - minValue
}

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()
    val n = br.readLine().toInt()
    val numberList = mutableListOf<Int>()
    repeat(n) {
        numberList.add(br.readLine().toInt())
    }
    bw.write("${getAverage(numberList)}\n")
    bw.write("${getMid(numberList)}\n")
    bw.write("${getFreq(numberList)}\n")
    bw.write("${getRange(numberList)}\n")
    bw.close()
    br.close()
}

더 좋은 풀이 방법 댓글로 달아주세요!!!