[문과 코린이의 IT 기록장] C++ 백준 문제풀이[DP] - 동물원 (1309)

[문과 코린이의 IT 기록장] C++ 백준 문제풀이[DP] - 동물원 (1309)

[문과 코린이의 IT 기록장] C++ 백준 문제풀이[DP]  - 동물원 (1309)

 

---

 

1309번: 동물원

[ 문제 ]

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

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

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

[ 입력 ]

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

[ 출력 ]

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

 

---

[ 코드 ]

#include<iostream>
using namespace std;

int dp[100001][3]; 
// 2*N배열에 사자를 배치하는 경우의 수
// dp[n][0] : n번째 줄에 사자가 배치되지 않는 경우의 수
// dp[n][1] : n번째 줄에 사자가 왼쪽에 배치되는 경우에 수
// dp[n][2] : n번째 줄에 사자가 오른쪽에 배치되는 경우의 수

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);

	int n; // 2*N배열
	cin >> n;

	dp[0][0] = 1; // 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 침
	// dp[0][1]과 dp[0][2]는 불가능한 일이므로 0으로 냅두기

	for (int i = 1; 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;
	}

	cout << (dp[n][0] + dp[n][1] + dp[n][2]) % 9901 << '\n';

	return 0;

}