[프로그래머스] 평균 구하기 Kotlin

안녕하세요. 이번에는 프로그래머스 평균 구하기를 풀어보려고 합니다.

 

https://school.programmers.co.kr/learn/courses/30/lessons/12944

프로그래머스

---

Problem

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

 

[제한 사항]

○ arr은 길이 1 이상, 100 이하인 배열입니다.

○ arr의 원소는 -10,000 이상 10,000 이하인 정수입니다. 

사진 1. 입출력 예

---

Solution

class Solution {
    fun solution(arr: IntArray): Double {
        var answer: Double = 0.0
        
        for(value in arr)
            answer += value
        
        return answer / arr.size
    }
}

 for문을 사용하여 arr의 값을 모두 더한 후 배열의 사이즈를 나눠주는 방법으로 문제를 접근하였다. 그런데 코틀린에서는 평균을 구할 수 있는 average() 함수를 사용하면 더 쉽게 문제를 풀 수 있다. 

 

1. Average 함수를 사용한 풀이

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

---

Performance

1. 내가 작성한 코드

 

2. Average() 함수를 사용한 코드