백준 6986 절사평균 Kotllin (정렬)

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

6986번: 절사평균

문제

체조나 다이빙 등의 경기에서 일부 심판이 자기가 좋아하는 선수에게 높은 점수를, 싫어하는 선수에게 낮은 점수를 주는 경우가 종종 있었다. 따라서 심판들이 주는 점수의 평균점수를 선수에게 주게 되면 공정하지 않은 경우가 생길 수 있다. 이를 방지하기 위하여 절사평균이나 보정평균을 사용한다. 예를 들어 심사위원 일곱 명이 다음과 같이 점수를 주었다고 하자.

9.3, 9.5, 9.6, 9.8, 9.1, 5.0, 9.3

전체의 합이 61.6이 되므로 평균은 8.8이 된다. 이 평균점수는 한 심판이 다른 심판에 비하여 아주 낮은 점수인 5.0을 주어서 나온 결과로, 선수는 매우 불공정하다고 느낄 것이다.

위의 점수를 작은데서 큰 순서로 정렬하면 5.0, 9.1, 9.3, 9.3, 9.5, 9.6, 9.8 이 된다.

이때 절사평균(7, 2)는 정렬된 전체 점수 일곱 개 중 양쪽 끝에서 두 개씩을 제외하고 난 9.3, 9.3, 9.5의 평균인 9.37이 된다(소수점이하 셋째 자리에서 반올림). 또 보정평균(7, 2)는 정렬된 전체 점수 일곱 개 중 양쪽 끝에서 각각 두 개를, 남은 점수 중 가장 가까운 것으로 교체한 9.3, 9.3, 9.3, 9.3, 9.5, 9.5, 9.5의 평균으로 9.39가 된다(소수점이하 셋째 자리에서 반올림).

N개의 점수와 양쪽에서 제외하는 개수 K 값이 주어졌을 때 절사평균(N, K)와 보정평균(N, K)를 계산하는 프로그램을 작성하시오.

입력

첫째 줄에 전체 점수의 개수 N과 제외되는 점수의 개수 K가 빈칸을 사이에 두고 주어진다. N은 3 이상 100,000 이하의 자연수이다. K는 0 이상 (N/2)-1 이하로 주어진다. 그 다음 N줄에는 각 심판의 점수가 한 줄에 하나씩 주어진다. 점수는 0 이상 10 이하의 실수로 소수점이하 첫째 자리까지 주어진다.

출력

첫째 줄에 절사평균(N, K)를, 둘째 줄에 보정평균(N, K)를 각각 소수점이하 셋째 자리에서 반올림하여 둘째 자리까지 출력한다. 예를 들어 결과값이 9.667인 경우 9.67로, 5인 경우 5.00으로, 5.5인 경우에는 5.50으로 출력한다.

알고리즘 분류

• 구현
• 정렬

풀이

단순 구현, 정렬 문제이다.

문제에 풀이가 다 나와있다.

우선 주어진 입력을 정렬하고 slice함수를 이용해 앞 뒤 각각 k개 만큼 자른 배열을 만든다. (절사평균)

그리고 정렬된 원본 입력에 대해 0부터 k개는 arr[k]로 바꾸고, n-k-1개부터 n-1개 까지는 arr[n-1-k]로 바꾼다. (보정평균)

이제 이를 String.format, round를 이용해서 소수점 셋째 자리에서 반올림하여 출력하면 끝

 

코드

import kotlin.math.roundToInt

val br = System.`in`.bufferedReader()
fun getIntList() = br.readLine().trim().split(' ').map { it.toInt() }

fun Double.formatter() = String.format("%.2f", (this * 100).roundToInt() /100.0)

fun main() = with(System.out.bufferedWriter()){

    //input
    val (n,k) = getIntList()
    val scoreArr = DoubleArray(n){br.readLine().toDouble()}
    //solve
    scoreArr.sort()
    for(i in 0 until k){
        scoreArr[i] = scoreArr[k]
    }
    for(i in n-1 downTo n-1-k){
        scoreArr[i] = scoreArr[n-1-k]
    }
    //output
    write("${scoreArr.slice(k until n-k).average().formatter()}\n")
    write(scoreArr.average().formatter())


    close()
}