Baekjoon 1309
in Study / Computer science on Baekjoon 1309
dynamic programming
Description
어떤 동물원에 가로로 두칸 세로로 N칸인 아래와 같은 우리가 있다.
이 동물원에는 사자들이 살고 있는데 사자들을 우리에 가둘 때, 가로로도 세로로도 붙어 있게 배치할 수는 없다. 이 동물원 조련사는 사자들의 배치 문제 때문에 골머리를 앓고 있다.
동물원 조련사의 머리가 아프지 않도록 우리가 2*N 배열에 사자를 배치하는 경우의 수가 몇 가지인지를 알아내는 프로그램을 작성해 주도록 하자. 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 친다고 가정한다.
Input
첫째 줄에 우리의 크기 N(1≤N≤100,000)이 주어진다.
Output
첫째 줄에 사자를 배치하는 경우의 수를 9901로 나눈 나머지를 출력하여라.
Example I & O
Input 1
4
Output 1
41
Input 2
5
Output 2
99
Input 3
100000
Output 3
8379
My first solution (memory overflow)
N = int(input())
dp = [0 for i in range(N + 1)]
dp[0] = 1
dp[1] = 3
for i in range(2, N + 1):
dp[i] = 2 * dp[i - 1] + dp[i - 2]
print(dp[N] % 9901)
Values of dp table are so big when N is big
My second solution
N = int(input())
dp = [0 for i in range(N + 1)]
dp[0] = 1
dp[1] = 3
for i in range(2, N + 1):
dp[i] = (2 * dp[i - 1] + dp[i - 2]) % 9901
print(dp[N])
the reason of 9901 existing in the problem
- Sketch
find the simple rule
Best answer
answer
Two solutions