1 분 소요

1. 문제

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

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

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

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

// 입력
첫째 줄에 우리의 크기 N(1N100,000) 주어진다.

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

// 예제 입력 1 
4

// 예제 출력 1 
41


2. 핵심 아이디어

  • 선택 안하는 경우 : 0 = 0 + 1 + 2
  • 왼쪽 타일 선택하는 경우 : 1 = 0 + 2
  • 오른쪽 타일 선택하는 경우 : 2 = 0 + 1


3. Python 문제풀이

import sys
input = sys.stdin.readline

n = int(input())
dp = [[0] * 3 for i in range(100001)]

for i in range(3) :
  dp[1][i] = 1
  
for i in range(2, 100001) :
  dp[i][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] + dp[i-1][2] % 9901
  
print(sum(dp[n]) % 9901)


4. Java 문제풀이


댓글남기기