본문 바로가기
코딩테스트/프로그래머스

[프로그래머스] 정렬 : K번째수 Kotlin

by 일상 속 둔치 2020. 9. 26.

배열의 구간을 자르고 해당 부분 배열을 정렬하는 문제이다.

 

Kotlin에 있는 slice라는 Array 내장 함수를 이용해 잘라주고 sort를 사용해 정렬한다!

* sort()는 객체를 정렬하고 return하지 않는다 (unit), sorted()는 정렬된 객체를 return한다.

 

그리고 인덱스에 맞는 값을 return 해준다

 

어렵지 않은 문제이고 정렬을 사용할 수 있는가를 물어보는 것 같다.

 

class Solution {
    fun solution(array: IntArray, commands: Array<IntArray>): IntArray {
        var answer = IntArray(commands.size)

        for(i in 0..answer.size-1){
            var tempArray = array.slice(commands[i][0]-1..commands[i][1]-1)
            tempArray = tempArray.sorted()
            answer[i] = tempArray[commands[i][2]-1]
        }

        return answer
    }
}

댓글