[Algorithm] 프로그래머스 - 뒤에서 5등까지 (Kotlin)

정수 리스트 num_list가 주어진다. num_list에서 가장 작은 5개의 수를 오름차순으로 담은 리스트를 리턴하는 함수를 완성하라

 

 

가장 작은 수 5개를 뽑아야 하고, 이 숫자들을 다시 오름차순으로 정렬해서 작은 숫자부터 표시되도록 하면 된다.

그렇다면 처음부터 리스트 안의 요소들을 오름차순 정렬한 다음, 5개를 뽑아서 배열로 만들어 리턴시키는 방식으로 풀 수 있다.

 

class Solution {
    fun solution(numList: IntArray): IntArray = numList.sorted().take(5).toIntArray()
}

 

또는 slice를 사용해서도 풀 수 있다.

 

class Solution {
    fun solution(num_list: IntArray): IntArray = num_list.sorted().slice(0..4).toIntArray()
}