2231번: 분해합

문제 출처 : https://www.acmicpc.net/problem/2231

2231번: 분해합

언어 : Kotlin

 

문제 설명 :

어떤 자연수 N이 있을 때, 그 자연수 N의 분해합은 N과 N을 이루는 각 자리수의 합을 의미한다. 어떤 자연수 M의 분해합이 N인 경우, M을 N의 생성자라 한다. 예를 들어, 245의 분해합은 256(=245+2+4+5)이 된다. 따라서 245는 256의 생성자가 된다. 물론, 어떤 자연수의 경우에는 생성자가 없을 수도 있다. 반대로, 생성자가 여러 개인 자연수도 있을 수 있다.

자연수 N이 주어졌을 때, N의 가장 작은 생성자를 구해내는 프로그램을 작성하시오.

• 첫째 줄에 자연수 N(1 ≤ N ≤ 1,000,000)이 주어진다.
• 첫째 줄에 답을 출력한다. 생성자가 없는 경우에는 0을 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 192MB

 

입출력 예 :

입력	출력
216	198

풀이 : 

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val n = br.readLine().toInt()
    var arr = emptyArray<Int>()

    for (i in n - 1 downTo 1) {
        val temp = i + i.toString().chunked(1).sumOf { it.toInt() }

        if (temp == n) arr += i
    }

    try {
        bw.write("${arr.min()}")
    } catch (_: Exception) {
        bw.write("0")
    }
    bw.flush()
    bw.close()
}