[Algorithm] 프로그래머스 - 배열 만들기 1 (Kotlin)

정수 n, k가 주어졌을 때 1 이상 n 이하의 정수 중에서 k의 배수를 오름차순으로 저장한 배열을 리턴하는 함수를 완성하라

 

 

주먹구구식으로 풀면 아래와 같다.

 

class Solution {
    fun solution(n: Int, k: Int): IntArray {
        val list = mutableListOf<Int>()
        (1 .. n).forEach {
            if (it % k == 0) {
                list.add(it)
            }
        }

        return list.toIntArray()
    }
}

 

간결하게 풀면 아래와 같이 할 수 있다.

 

class Solution {
    fun solution(n: Int, k: Int) = (k..n step k).toList()
}