[Python/Silver1/1309] 동물원

|    문제

어떤 동물원에 가로로 두칸 세로로 N칸인 아래와 같은 우리가 있다.

이 동물원에는 사자들이 살고 있는데 사자들을 우리에 가둘 때, 가로로도 세로로도 붙어 있게 배치할 수는 없다. 이 동물원 조련사는 사자들의 배치 문제 때문에 골머리를 앓고 있다.

동물원 조련사의 머리가 아프지 않도록 우리가 2*N 배열에 사자를 배치하는 경우의 수가 몇 가지인지를 알아내는 프로그램을 작성해 주도록 하자. 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 친다고 가정한다.

 

|    입력

첫째 줄에 우리의 크기 N(1≤N≤100,000)이 주어진다.

 

|    출력

첫째 줄에 사자를 배치하는 경우의 수를 9901로 나눈 나머지를 출력하여라 .

 

 

---

정답 코드

import sys
input =sys.stdin.readline

class Solution:
    def zoo(self):
        n = int(input())
        dp = [0] * (n+1)
        if n == 1:
            print(3)
            return
        dp[1], dp[2] = 3, 7
        for x in range(3, n+1):
            dp[x] = (2*dp[x-1] + dp[x-2]) % 9901
        print(dp[n])
        
if __name__ == "__main__":
    s = Solution()
    s.zoo()

 

 

|    비고

n=1일때부터 계산해보면, dp[x] = 2*dp[x-1] + dp[x-2] 라는 점화식을 얻을 수 있다. dp 문제는 항상 연습장이 필요한 것 같다.