동캄의 코딩도장

백준 1309 [동물원] 파이썬 본문

코테/BOJ

백준 1309 [동물원] 파이썬

동 캄 2022. 2. 9. 20:52

https://www.acmicpc.net/problem/1309

 

1309번: 동물원

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

www.acmicpc.net

N = int(input())
dp = [0, 3, 7]

for k in range(3, N+1):
    dp.append((dp[k-1]*2+dp[k-2])%9901)

print(dp[N])

dp[k-1]에는 k-1번째의 경우에 따라 칠할지 말지를 정할 수 있으므로,  x2를 해준다.

dp[k-1], dp[k-2] 모두 아예 안칠하는 경우도 존재하므로, dp[k-2]를 더해준다.