[백준] 설탕 배달 - 실버 4 (Kotlin)

문제

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.

입력

첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)

출력

상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.

풀이

A[N] 의 값은 N 킬로그램을 배달할 수 있는 최소 봉지의 갯수를 표현한다.
상향식 DP 를 사용하여 아래의 식을 만들었다.
A[N] = min (A[N-3] + 1, A[N-5] + 1)

해석

A[0] ~ A[4] 는 초기 값을 넣어준다.
A[0] = 0, A[1] = INF, A[2] = INF, A[3] = 1, A[4] = INF
A[5] = min (A[2] + 1, A[0] + 1) = min (INF + 1, 1) = 1
A[6] = min (A[3] + 1, A[1] + 1) = min (2, INF + 1) = 2
A[7] = min (A[4] + 1, A[2] + 1) = min (INF + 1, INF + 1) = INF + 1
A[8] = min (A[5] + 1, A[3] + 1) = min (2, 2) = 2
...
A[6], A[8] 의 값을 확인하면 실제로 6의 경우 3 봉지 2개로 일치한다는 것을 알 수 있고, 8의 경우에도 3 봉지 1개 + 5 봉지 1개 = 2개로 일치한다는 것을 알 수 있다.
따라서, N 까지 해당 값들을 계산한 뒤 A[N] 의 값을 출력해주면 답을 구할 수 있다.

소스 코드

const val INF = 987654321
fun main(args: Array<String>) {
    val N = readLine()!!.toInt()
    val dp = IntArray(5001)

    dp[0] = 0
    dp[1] = INF
    dp[2] = INF
    dp[3] = 1
    dp[4] = INF

    for (i in 5..N) {
        dp[i] = Math.min(dp[i - 3] + 1, dp[i - 5] + 1)
    }

    println(if (dp[N] >= INF) "-1" else dp[N])
}