[Algorithm] 프로그래머스 - 자릿수 더하기 (Kotlin)

정수 n이 매개변수로 주어질 때 n의 각 자리 숫자의 합을 리턴하는 solution()을 완성하라

 

 

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

 

class Solution {
    fun solution(n: Int): Int {
        val list = n.toString().split("")
        var result = 0
        for (i in list) {
            if (i != "") {
                result += i.toInt()
            }
        }

        return result
    }
}

 

이걸 줄여봤다.

 

class Solution {
    fun solution(n: Int) = n.toString().split("").filter { it != "" }.sumOf { it -> it.toInt() }
}

 

그런데 써야 하는 함수가 너무 많다. 더 줄일 순 없나 고민해도 모르겠어서 다른 사람의 풀이를 참고했다.

 

class Solution {
    fun solution(n: Int): Int = n.toString().toList().sumOf { it.digitToInt() }
}