[Algorithm] 프로그래머스 - 순서 바꾸기 (Kotlin)

정수 리스트 num_list, 정수 n이 주어질 때 num_list를 n번째 원소 이후의 원소들과 n번째까지의 원소들로 나눠 n번째 원소 이후의 원소들을 n번째까지의 원소들 앞에 붙인 리스트를 리턴하는 함수를 완성하라

 

 

쉽게 말해 n 이상의 인덱스에 있는 값들을 그대로 0번 인덱스 앞에 갖다 붙이라는 뜻이다. 2번째 입출력 예를 보면 순서 변경 없이 그대로 0번 인덱스 앞에 붙인 걸 볼 수 있다.

copyOfRange와 subList를 활용하면 풀 수 있다.

 

class Solution {
    fun solution(numList: IntArray, n: Int): IntArray {
        val after = numList.copyOfRange(n, numList.size).toList()
        val before = numList.toList().subList(0, n)

        return (after + before).toIntArray()
    }
}

 

리턴타입이 IntArray기 때문에 두 리스트를 더한 다음 toIntArray로 캐스팅해야 한다.