[백준][KOTLIN] 9461 파도반 수열

 

9461번: 파도반 수열

풀이

1. 주어진 1부터 10까지 결과에 대한 규칙을 찾는다 : P(N) = P(N - 2) + P(N - 3)
2. 입력이 100일 경우 Int Data 범위를 넘어서기 때문에 LongArray 를 배열로 선언한다.

코드

import java.io.*

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

    repeat(T) {
        val N = br.readLine().toInt()
        val dp = LongArray(N + 1)

        for (i in 1..N) {
            if (i == 1 || i == 2 || i == 3) {
                dp[i] = 1 
            } else {
                dp[i] = dp[i - 2] + dp[i - 3]   
            }
        }

        bw.write("${dp[N]}\n")
    }

    bw.flush()
    bw.close()
    br.close()
}