[백준] 1, 2, 3 더하기 (9095)

원본 문제

문제 설명

정수 4를 1, 2, 3의 합으로 나타내는 방법은 총 7가지가 있다. 합을 나타낼 때는 수를 1개 이상 사용해야 한다.

• 1+1+1+1
• 1+1+2
• 1+2+1
• 2+1+1
• 2+2
• 1+3
• 3+1

정수 n이 주어졌을 때, n을 1, 2, 3의 합으로 나타내는 방법의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다. n은 양수이며 11보다 작다.

출력

각 테스트 케이스마다, n을 1, 2, 3의 합으로 나타내는 방법의 수를 출력한다.

테스트 케이스

입력	출력
3 4 7 10	7 44 274

문제 풀이 (KOTLIN)(DP)

var dp = IntArray(12)

fun main() = with(System.`in`.bufferedReader()) {
    dp[1] = 1
    dp[2] = 2
    dp[3] = 4
    val cases = readLine().toInt()
    repeat(cases) {
        val num = readLine().toInt()
        println(dp(num))
    }
}

fun dp(n: Int): Int {
    return if (dp[n] != 0) dp[n]
    else {
        when (n) {
            1 -> 1
            2 -> 2
            3 -> 4
            else -> {
                dp[n] = dp(n - 1) + dp(n - 2) + dp(n - 3)
                dp[n]
            }
        }
    }
}

문제 풀이 (SWIFT)(DP)

let n = Int(readLine()!)!
var dp = [Int](repeating: 0, count: 12)
dp[1] = 1
dp[2] = 2
dp[3] = 4

for j in 4...11 {
    dp[j] = dp[j - 1] + dp[j - 2] + dp[j - 3]
}

for _ in 0..<n {
    print(dp[Int(readLine()!)!])
}