[Algorithm] 프로그래머스 - 약수 구하기 (Kotlin)

정수 n이 매개변수로 주어질 때, n의 약수를 오름차순으로 담은 배열을 리턴하는 solution()을 완성하라

 

 

주먹구구식으로 푼 풀이는 아래와 같다.

 

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

        return result.toIntArray()
    }
}

 

반복문 안에서 조건문 조건에서 실수하면 안 된다. 입력된 값을 1부터 n까지의 숫자로 나눠야 하기 때문이다.

이걸 줄이면 아래와 같이 된다.

 

class Solution {
    fun solution(n: Int) = (1..n).toList().filter { n % it == 0 }
}