[Algorithm] 프로그래머스 - n의 배수 고르기 (Kotlin)

정수 n, 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 리턴하는 solution()을 완성하라

 

 

아래는 주먹구구식으로 푼 풀이다. 귀찮아서 리스트 하나 더 만들고 여기에 배수인 숫자들만 들어가게 했다.

 

class Solution {
    fun solution(n: Int, numlist: IntArray): IntArray {
        val list = numlist.toMutableList()
        val result = arrayListOf<Int>()
        for (i in list) {
            if (i % n == 0) {
                result.add(i)
            }
        }

        return result.toIntArray()
    }
}

 

이걸 줄이면 아래와 같다.

 

class Solution {
    fun solution(n: Int, numlist: IntArray) = numlist.filter { it % n == 0 }
}