Kotlin - 평균 구하기 프로그래머스 코딩 연습 (12)

프로그래머스 문제

정수를 담고 있는 배열 arr의 평균값을 return하는 함수, solution을 완성해보세요.

arr	return
[1,2,3,4]	2.5
[5,5]	5

코틀린 답

답

class Solution {
    fun solution(arr: IntArray): Double {
        if (arr.isEmpty()) {
            return 0.0
        }

        val sum = arr.sum()
        return sum.toDouble() / arr.size
    }
}

답

class Solution {
    fun solution(arr: IntArray): Double 
    {
        return arr.average()
    }
}

풀이

1번 답

arr 라는 이름의 'IntArray'를 받으며, 'Double' 타입의 값을 반환합니다.

arr이 비어있는지 .isEmpty()를 통하여 확인하고 요소가 비어있을 경우, 함수는 0.0을 반환합니다. 이유는 배열이 비어 있으면 계산할 수 없기 때문입니다.

arr의 모든 요소의 합계를 계산하여 'sum'에 저장합니다.

마지막으로 배열의 합계를 toDouble()를 사용하여, Double타입으로 반환 후 'arr.size'로 나누어 평균값을 계산합니다.

• 'arr.size'는 val arr = intArrayOf(1, 2, 3, 4, 5) 이라면 5가 됩니다.

2번 답

코틀린에서는 'average()' 함수를 사용하여 평균값을 계산할 수 있습니다.

arr 라는 이름의 'IntArray'를 받으며, 'Double' 타입의 값을 반환하고 그 값으로 .average() 를 통해 식없이 바로 계산됩니다.

반응형