[Programmers] Lv1. K번째 수(kotlin)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42748?language=kotlin

 

 

예시

https://school.programmers.co.kr/learn/courses/30/lessons/42748?language=kotlin

 

 

풀이

1. return 값의 갯수는 commands 수만큼 결과가 저장이 됩니다.

2. commands의 index를 하나씩 탐색하여 List로 변경 후 subList 함수로 부분 리스트를 가져옵니다.

3. 이후 sorted로 정렬한 리스트를 가져온 다음에 원하는 인덱스의 값을 찾으면 됩니다.

 

class Solution {
    fun solution(array: IntArray, commands: Array<IntArray>): IntArray {
        return commands.mapIndexed{ index, i ->
            array.toList()
                    .subList(commands[index][0] - 1, commands[index][1])
                    .sorted()
                    .get(commands[index][2] -1)
        }.toIntArray()
    }
}

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/-list/sub-list.html

subList - Kotlin Programming Language