[Algorithm] 프로그래머스 - 합성수 찾기 (Kotlin)

약수 개수가 3개 이상인 수를 합성수라고 한다. 자연수 n이 매개변수로 주어질 때, n 이하의
합성수 개수를 리턴하는 solution()을 완성하라

 

 

약수를 구하고 약수 개수가 3개 이상인 경우 리턴하는 것까지는 구현했으나 이상하게 자꾸 에러가 발생하면서 몇 개는 맞고 몇 개는 틀렸다고 나왔다.

고민하다가 찾아보니 매개변수가 1이 들어올 수도 있기 때문에 이 경우까지 모두 처리를 해야만 작동했다. 아래는 참고한 자바 풀이 링크와 코틀린으로 바꾼 코드다.

 

https://mycodingreview.tistory.com/entry/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4%ED%95%A9%EC%84%B1%EC%88%98%EC%B0%BE%EA%B8%B0

[프로그래머스/Java] Lv.0 합성수 찾기

class Solution {
    fun div(n: Int): Int {
        var count = 0
        for (i in 1 .. n) {
            if (n % i == 0) count++
        }

        return count
    }

    fun solution(n: Int): Int {
        var answer = 0
        for (i in 1 .. n) {
            if (div(i) != 2 && div(i) != 1) {
                answer++
            }
        }
        return answer
    }
}

 

정답을 확인하고 다른 풀이를 확인해 보니 역시 간단하게 구현한 코드가 있었다.

 

class Solution {
    fun solution(n: Int) = (1..n).filter { i -> (1..i).filter { i % it == 0 }.size > 2 }.size
}