[Algorithm] 프로그래머스 - n번째 원소부터 (Kotlin)

정수 리스트 num_list와 정수 n이 주어질 때, n번째 원소부터 마지막 원소까지의 모든 원소를 담은 리스트를 리턴하는 함수를 완성하라

 

 

n번째 원소~마지막 원소를 모두 가진 리스트를 만들면 되는데, copyOfRange()를 쓰면 단번에 해결된다.

 

https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/copy-of-range.html

copyOfRange - Kotlin Programming Language

원본 배열의 지정된 범위의 복사본인 새 배열을 리턴한다
- fromIndex : 복사할 범위의 시작 부분(inclusive)
- toIndex : 복사할 범위의 끝(exclusive)

 

리턴되는 배열은 시작값, 끝 값을 모두 포함하기 때문에 이걸 유의하며 작성하면 된다.

 

class Solution {
    fun solution(numList: IntArray, n: Int) = numList.copyOfRange(n - 1, numList.size)
}

 

 

문제에선 n번째 원소라고 적혀 있지만 입출력 예를 보면 n-1번째 원소~마지막 원소를 담고 있다.

이 함정에 빠지지 않는다면 크게 어렵지 않게 풀 수 있다.