[Silver I] 동물원 - 1309

문제 링크

성능 요약

메모리: 17712 KB, 시간: 140 ms

분류

다이나믹 프로그래밍

제출 일자

2024년 2월 6일 10:06:01

문제 설명

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

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

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

입력

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

출력

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

현재 코드(JAVA)

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

public class Main {
	static int[][] dp;
	public static void main(String[] args) throws IOException {
		BufferedReader br =new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		dp=new int[n+1][3];
		dp[1][0]=1; dp[1][1]=1; dp[1][2]=1;
		//Arrays.fill(dp[1],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 ans=0;
        
		for(int i=0; i<=2; i++) {
			ans+=dp[n][i];
		}
        
        ans %=9901;
		System.out.println(ans);
	}
}

이전 코드(C++)

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main() {
	long long dp[100001];
	int n;
	cin >> n;
	dp[1] = 3;
	dp[2] = 7;
	dp[3] = 17;
	dp[4] = 41;
	for (int i = 5; i <= n; i++) {
		dp[i] = (dp[i - 2] + dp[i - 1] * 2) % 9901;
	}
	cout << dp[n] << endl;
 }

이전에 풀었던 내용을 수정해서 다시 풀었다.
현재 코드가 2차원 배열을 사용해 조금 더 이해하기 쉬운 것 같다.
[문제 풀이 참고]