[백준] 수찬은 마린보이야!! (15921)(kotlin)

원본 문제

문제 설명

부산의 자랑! 동래의 희망! 부산 앞바다에서 수영선수의 꿈을 키워온 수찬이는 우리나라(자투리 지식: 우리 한민족이 세운 나라를 스스로 이르는 말은 합성어로서 ‘우리나라’로 붙여 표기)의 자랑스런 국가대표 수영선수가 되었다.

수찬이는 인터넷 뉴스를 통해 이번 2020 도쿄 하계올림픽 관련 전문가들의 인터뷰를 찾아보고 있다. 자신을 교수라고 소개한 어떤 전문가에 따르면, 유럽의 유명 대학 연구팀의 연구 결과에 따르면, 통계적으로 어떤 선수가 대회에서 연습과 비슷하게 꾸준한 기량을 뽐낼 수 있는 확률은 (연습 기록들의 평균값) / (연습 기록들 중 하나를 균일한 확률로 뽑을 때의 기댓값) 이라고 한다.

수찬이는 이 기사를 읽고 무릎을 탁(!) 치며 자신의 연습 기록을 가지고 이 확률을 계산해보기로 했다. 수찬이가 꾸준한 기량을 뽐낼 수 있는 확률은 얼마일까?

입력

첫째 줄에 수찬이의 연습 기록의 개수 N이 주어진다. (0 ≤ N ≤ 100)

둘째 줄에 수찬이의 연습 기록 N개가 주어진다. N이 0이면 아무것도 주어지지 않으며 연습 기록은 100 이하의 양의 정수이다.

출력

문제에서 제시한 확률을 소수 셋째 자리에서 반올림하여 소수 둘째 자리까지 출력한다. N이 0이거나 분모가 0인 경우에는 “divide by zero”를 출력한다.

테스트 케이스

입력	출력
5 5 10 10 15 20	1.00
0	divide by zero

힌트

기댓값 E(X)의 정의는 ‘각 사건이 벌어졌을 때의 이득과 그 사건이 벌어질 확률을 곱한 것을 전체 사건에 대해 합한 값’이다. 다시 말해, 어떤 수 x가 수열에 등장할 확률 P(x) = (x의 등장 횟수) / (전체 수열의 길이)이고, 이때 기댓값은 수열에 등장하는 모든 x에 대해 (x * P(x))를 구해 모두 더한 값이 된다.

$E(X)=\sum _i{p}_i{x}_i$

예시1을 보자. 평균은 (5+10+10+15+20)/5로 12이며, 기댓값은 5(1/5) + 10(2/5) + 15(1/5) + 20(1/5)로 역시 12이다. 따라서 확률은 12/12로 1이다.

문제 풀이1

fun main(args: Array<String>) {
    val n = readLine()!!.toInt()

    if (n == 0) {
        println("divide by zero")
        return
    }

    var average = 0.0
    val practices = readLine()!!
        .split(" ")
        .map { it.toDouble() }
        .also { average = it.reduce { acc, d -> acc + d } / 5.0 }
        .groupingBy { it }
        .eachCount()
        .map { it.key * it.value / 5.0 }
        .reduce { acc, i -> acc + i }

    println(String.format("%.2f", average/practices))
}

문제 풀이2

fun main(args: Array<String>) {
    println(
        if (readLine()!! == "0") "divide by zero"
        else "1.00"
    )
}