관리 메뉴

진취적 삶

동물원 본문

알고리즘/다이나믹 프로그래밍 연습

동물원

hp0724 2023. 6. 30. 12:54

bookmark

문제

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

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

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

입력

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

출력

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

모범답안

새로추가되는 우리에 사자를 넣은 경우 와 안넣는 경우를 나눠서 넣으면된다.

import sys

n = int(sys.stdin.readline())
MOD = 9901
dp = [0] * (n + 1)
dp[0] = 1
dp[1] = 3
for i in range(2, n + 1):
    dp[i] = (dp[i - 1] * 2 + (dp[i - 2] * 1)) % MOD

print(dp[n])

'알고리즘 > 다이나믹 프로그래밍 연습' 카테고리의 다른 글

타일 채우기  (0) 2023.07.02
1,2,3 더하기 3  (0) 2023.06.30
RGB 거리  (0) 2023.06.30
오르막수  (0) 2023.06.30
스티커  (0) 2023.06.30