1309번 - 동물원

하우르·2021년 6월 4일
0

백준인강

목록 보기
24/30

문제

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

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

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

입력

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

출력

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

예제 입력 1

4

예제 출력 1

41

풀이

우리의 크기 N(1≤N≤100,000)
D[N] = 2*N 배열에 사자를 배치하는 경우의 수

  • D[N][0] = N번 줄에 배치하지 않음
  • D[N][1] = N번 줄의 왼쪽에만 배치함
  • D[N][2] = N번 줄의 오른쪽에만 배치함
    D[1][0] = 1
    D[1][1] = 1
    D[1][2] = 1

처음에 틀렸었는데
D[N][0]일 경우 그 전 D[N-1][0]도 포함 된다는 것을 깜빡함

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

public class Main {
	public static void main(String args[]) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
		// 우리의 크기 N(1≤N≤100,000)
		int N = Integer.parseInt(reader.readLine());
		int mod = 9901;

		long[][] d = new long[N][3];
		d[0][0] = 1;
		d[0][1] = 1;
		d[0][2] = 1;
		for (int i = 1; i < N; i++) {
			d[i][0] = (d[i - 1][0] + d[i - 1][1] + d[i - 1][2])%mod;
			d[i][1] = (d[i - 1][0] + d[i - 1][2])%mod;
			d[i][2] = (d[i - 1][0] + d[i - 1][1])%mod;
		}
		System.out.println((d[N-1][0]+d[N-1][1]+d[N-1][2])%mod);
	}
}
profile
주니어 개발자

0개의 댓글