C++ 백준 1309번

다이나믹 프로그래밍!!!!!

백준 1309번 www.acmicpc.net/problem/1309

1309번: 동물원

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

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

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

 

동물들이 가로로 붙어있을수 없기 때문에

한 행에 동물을 놓을 수 있는 방법은 o x , x o, x x 3가지 방법이 있고 이걸 편하게 0, 1, 2라고 부르겠다.

마지막 행에 0이 오면 그 앞에 행에는 1,2가 올 수 있고(0이 오면 세로로 붙어버림!)

마지막 행에 1이오면 그 앞에 행에는 0,2가 올 수 있다.

마지막 행에 2가 오면 앞에 어떤게 있든 세로로 붙지 않으니까 0,1,2 다 가능하다.

 

d[n][i] = n번째 행에 i방법의 동물을 놓는 경우의 수라고 하면

d[n][0] = d[n-1][1] + d[n-1][2]

d[n][1] = d[n-1][0] + d[n-1][2]

d[n][2] = d[n-1][0] + d[n-1][1] + d[n-1][2]

로 표현할 수 있다!!

 

#include<iostream>
using namespace std;
int d[100000][3];
int mod = 9901;
int main() {
	int n;
	cin >> n;
	d[1][0] = 1;
	d[1][1] = 1;
	d[1][2] = 1;
	for (int i = 2; i <= n; i++) {
		d[i][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] + d[i - 1][2]) % mod;
	}
	cout << (d[n][0] + d[n][1] + d[n][2]) % mod << '\n';
}