[백준]#1309 동물원

문제

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

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

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

입력

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

출력

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

예제 입력 1

4

예제 출력 1

41

풀이

이 문제는 DP 알고리즘을 이용해서 풀면 되는데 DP 알고리즘은 점화식 세우기가 참 힘겹다...

그리고 문제 출력 조건을 잘 보는게 중요하다. 가끔 문제를 대충 봐서 출력 조건에서 틀릴때가 있다.

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

public class Main {
    static long[] dp;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        if(N==1)
            System.out.println(3);

        else if(N==2)
            System.out.println(7);

        else {
            dp = new long[N+1];
            DP(N);
            System.out.println(dp[N]);
        }
    }

    static void DP(int n) {
        dp[1]=3;
        dp[2]=7;

        for(int i=3; i<=n; i++) {
            dp[i] = (2*dp[i-1] + dp[i-2])%9901;
        }
    }
}