[백준 11727] 2 * n 타일링2

문제

링크

2×n 직사각형을 1×2, 2×1과 2×2 타일로 채우는 방법의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 n이 주어진다. (1 ≤ n ≤ 1,000)

출력

첫째 줄에 2×n 크기의 직사각형을 채우는 방법의 수를 10,007로 나눈 나머지를 출력한다.

풀이

앞에 타일링 문제에서 2 * 2 타일이 추가되었을 뿐 개념은 같다 (n - 1) + (n - 2) * 2 > 2 * 2 타일은 결국 ( n - 2 ) 타일과 같다

func solve(n : Int) -> Int {
    if n == 0 { return 0}
    if n == 1 { return 1}
    if n == 2 { return 3}
    var arr = [Int](repeating: 0, count : n + 1)

    arr[1] = 1
    arr[2] = 3
    for i in 3 ... n {
        arr[i] = arr[i - 1] + (arr[i - 2] * 2)
        arr[i] %= 10007
    }
    return arr[n]
}