[Java] 백준 1309번 : 동물원

문제

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

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

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

입력

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

출력

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

예제 입력 1

4

예제 출력 1

41

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

	private static BufferedReader br;
	
	private static int N;
	private static int [][]dp;
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		br=new BufferedReader(new InputStreamReader(System.in));
		N=Integer.parseInt(br.readLine());
		dp=new int[N+1][3];	// 0: 00, 1: 01, 2: 10

		dp[1][0]=1;
		dp[1][1]=1;
		dp[1][2]=1;
		for(int i=2;i<=N;i++) {
			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;
		}
		
		int result=(dp[N][0]+dp[N][1]+dp[N][2])%9901;
		System.out.println(result);
	}

}

i번째 줄까지 가능한 경우의 수를 dp배열에 저장해주었습니다.

이때 이전 줄에서 가능한 가짓수 00, 01, 10 ->세 가지로 구분하여 계산해주었습니다.

이전 줄이 다음과 같을 때 현재 줄에서 가능한 경우

00 -> 00, 01, 10

01 -> 00, 10

10 -> 00, 01

 

각 계산마다 9901로 나눈 나머지를 저장해주며 N번째 줄에서의 최종 경우의 수를 구하였습니다.