[백준][KOTLIN] 10844 쉬운 계단 수

https://www.acmicpc.net/problem/10844

10844번: 쉬운 계단 수

풀이

코드

import java.util.*

fun main(args: Array<String>) {
    val sc = Scanner(System.`in`)
    val N = sc.nextInt()
    val dp = Array(N + 1) {
        IntArray(10)
    }

    for (j in 1..9) {
        dp[1][j] = 1
    }

    for (i in 2..N) {
        for (j in 0..9) {
            dp[i][j] = when (j) {
                0 -> dp[i - 1][j + 1] % 1_000_000_000
                9 -> dp[i - 1][j - 1] % 1_000_000_000
                else -> (dp[i - 1][j - 1] + dp[i - 1][j + 1]) % 1_000_000_000
            }
        }
    }

    var sum = 0

    for (i in 0..9) {
        sum += dp[N][i]
    }

    println("$sum")
}