[Lv. 0] n개 간격의 원소들

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181888

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 마지막 원소까지 n개 간격으로 저장되어있는 원소들을 차례로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

제한 사항 : 

• 5 ≤ num_list의 길이 ≤ 20
• 1 ≤ num_list의 원소 ≤ 9
• 1 ≤ n ≤ 4

입출력 예 :

num_list	n	result
[4, 2, 6, 1, 7, 6]	2	[4, 6, 7]
[4, 2, 6, 1, 7, 6]	4	[4, 7]

풀이 : 

class Solution {
    fun solution(num_list: IntArray, n: Int): IntArray {
        var answer: IntArray = intArrayOf()

        for (i in num_list.indices step n) {
            answer += num_list[i]
        }

        return answer
    }
}

 

더 짧게 만든다면 이런 것도 가능할 것이다.

class Solution {
    fun solution(num_list: IntArray, n: Int): IntArray = num_list.slice(0..num_list.lastIndex step n).toIntArray()
}