[알고리즘] 백준 1309번 동물원 :: 우유

문제

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

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

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

입력

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

출력

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

---

경우의 수를 3가지로 나누어 생각해볼 수 있다.

1) n번째 줄에 사자가 왼쪽에 있는경우 

n-1번째 줄의 오른쪽 칸에 사자가 있거나 + 양쪽 칸에 사자가 모두 없는 경우

DP[i][1] = (DP[i - 1][0] + DP[i - 1][2])

 

2) n번째 줄에 사자가 오른쪽에 있는경우 

n-1번째 줄의 왼쪽 칸에 사자가 있거나 + 양쪽 칸에 사자가 모두 없는 경우

DP[i][2] = (DP[i - 1][0] + DP[i - 1][1])

 

3) n번째 줄에 사자가 없는 경우 

n-1번째 줄에 왼쪽 오른쪽 모두 사자가 없는 경우 + 왼쪽에 있는 경우 + 오른쪽에 있는 경우

D[DP[i][0] = (DP[i - 1][0] + DP[i - 1][1] + DP[i - 1][2])

 

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

// BOJ_1309
// 동물원
// DP
public class Main {

	static int N;
	static int DP[][];

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());

		N = Integer.parseInt(st.nextToken());

		int[][] DP = new int[N + 1][3];

		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;
		}
		System.out.println((DP[N][0] + DP[N][1] + DP[N][2]) % 9901);
	}
}