백준 10844 쉬운 계단수

문제

45656이란 수를 보자.
이 수는 인접한 모든 자리의 차이가 1이다. 이런 수를 계단 수라고 한다.
N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구해보자. 0으로 시작하는 수는 계단수가 아니다.

입력

첫째 줄에 N이 주어진다. N은 1보다 크거나 같고, 100보다 작거나 같은 자연수이다.

출력

첫째 줄에 정답을 1,000,000,000으로 나눈 나머지를 출력한다.

import java.io.*;
import java.util.*;

public class Main {
    static long[][] dp; // dp[i][j] 길이가 i일때 마지막 자리가 j일 때 계단수의 개수
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        dp = new long[N + 1][10];
        Arrays.fill(dp[1], 1);
        dp[1][0] = 0;
        for (int i = 2; i <= N; i++) {
            for (int j = 0; j < 10; j++) {
                if(j == 0) dp[i][0] = dp[i - 1][1] % 1000000000;
                else if(j == 9) dp[i][j] = dp[i - 1][j - 1] % 1000000000;
                else dp[i][j] = (dp[i - 1][j - 1] + dp[i - 1][j + 1]) % 1000000000;
            }
        }
        long sum = 0;
        for (int i = 0; i < 10; i++) {
            sum += dp[N][i];
        }
        bw.write(sum % 1000000000 + "\n");
        bw.flush();
        bw.close();
    }
}

dp문제이다.
손으로 예시를 몇 가지 써보면 점화식을 쉽게 떠올릴 수 있었던 문제.
long으로 설정하는 걸 잊지 말자!