문제
어떤 동물원에 가로로 두칸 세로로 N칸인 아래와 같은 우리가 있다.
이 동물원에는 사자들이 살고 있는데 사자들을 우리에 가둘 때, 가로로도 세로로도 붙어 있게 배치할 수는 없다. 이 동물원 조련사는 사자들의 배치 문제 때문에 골머리를 앓고 있다.
동물원 조련사의 머리가 아프지 않도록 우리가 2*N 배열에 사자를 배치하는 경우의 수가 몇 가지인지를 알아내는 프로그램을 작성해 주도록 하자. 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 친다고 가정한다.
입력
첫째 줄에 우리의 크기 N(1≤N≤100,000)이 주어진다.
출력
첫째 줄에 사자를 배치하는 경우의 수를 9901로 나눈 나머지를 출력하여라.
코드
n = int(input())
dp = [[0]*3 for _ in range(n)]
dp[0][0] = 1 # 사자 배치 X
dp[0][1] = 1 # 0열에 사자 배치
dp[0][2] = 1 # 1열에 사자 배치
for i in range(1, n):
dp[i][0] = (dp[i-1][0] + dp[i-1][1] + dp[i-1][2]) % 9901
dp[i][1] = (dp[i-1][0] + dp[i-1][2]) % 9901
dp[i][2] = (dp[i-1][0] + dp[i-1][1]) % 9901
ans = (dp[n-1][0] + dp[n-1][1] + dp[n-1][2]) % 9901
print(ans)
DP를 이용하여 각 행에 1. 사자를 배치 안했을때, 2. 0열에 사자 배치했을때, 3. 1열에 사자 배치했을때의 3가지 경우로 나누어 경우의 수를 구한다.
만약 메모리 초과가 발생한다면 9901로 나눠주는 것을 잊지는 않았는지 확인하자.
'문제풀이 > DP' 카테고리의 다른 글
[Python/파이썬] 백준 1958번 LCS 3 (0) | 2024.01.05 |
---|---|
[Python/파이썬] 백준 11060번 점프 점프 (0) | 2023.12.25 |
[Python/파이썬] 백준 1535번 안녕 (0) | 2023.11.30 |
[Python/파이썬] 백준 14699번 관악산 등산 (2) | 2023.11.22 |
[Python/파이썬] 백준 1495번 기타리스트 (1) | 2023.11.20 |