알고리즘 문제 풀이/프로그래머스
[Algorithm] 프로그래머스 - n번째 원소부터 (Kotlin)
참깨빵위에참깨빵_
2024. 4. 20. 20:34
728x90
반응형
정수 리스트 num_list와 정수 n이 주어질 때, n번째 원소부터 마지막 원소까지의 모든 원소를 담은 리스트를 리턴하는 함수를 완성하라
n번째 원소~마지막 원소를 모두 가진 리스트를 만들면 되는데, copyOfRange()를 쓰면 단번에 해결된다.
https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/copy-of-range.html
copyOfRange - Kotlin Programming Language
kotlinlang.org
원본 배열의 지정된 범위의 복사본인 새 배열을 리턴한다
- fromIndex : 복사할 범위의 시작 부분(inclusive)
- toIndex : 복사할 범위의 끝(exclusive)
리턴되는 배열은 시작값, 끝 값을 모두 포함하기 때문에 이걸 유의하며 작성하면 된다.
class Solution {
fun solution(numList: IntArray, n: Int) = numList.copyOfRange(n - 1, numList.size)
}
문제에선 n번째 원소라고 적혀 있지만 입출력 예를 보면 n-1번째 원소~마지막 원소를 담고 있다.
이 함정에 빠지지 않는다면 크게 어렵지 않게 풀 수 있다.
반응형