[JAVA] 백준 10844번

[JAVA] 백준 10844번 - 쉬운 계단 수

 

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

public class Main {

	static int[][] dp = new int[101][10];

	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());
		int answer = 0;
		int mod = 1000000000;

		// dp[i][0] = 마지막 자리가 0인경우
		// dp[i][1] = 마지막 자리가 1인경우
		// ...
		// dp[i][9] = 마지막 자리가 9인경우

		for (int i = 1; i < 10; i++) { // 처음 값 세팅
			dp[1][i] = 1;
		}

		for (int i = 2; i <= N; i++) { // N번째까지 구하기

			for (int j = 0; j < 10; j++) {
				if (j == 0) { // 마지막 자리가 0인 경우 -> N-1번째에서 마지막자리가 1인 개수
					dp[i][j] = dp[i - 1][1] % mod;
				} else if (j == 9) { // 마지막 자리가 9인 경우 -> N-1번째에서 마지막자리가 8인 개수
					dp[i][j] = dp[i - 1][8] % mod;
				} else { // 마지막 자리가 1~8인 경우
					dp[i][j] = (dp[i - 1][j - 1] + dp[i - 1][j + 1]) % mod;
				}
			}

		}

		for (int num : dp[N]) {
			answer = (answer + num) % mod;
		}

		bw.write(String.valueOf(answer));
		bw.flush();
		bw.close();

	}
}

정답을 맞춘 풀이방법

1. 마지막 자리수에 따라 생성되는 숫자의 개수를 저장함

2. 마지막 자리수가 0인경우와 9인 경우는 1개의 숫자만을 생성가능, 2~8인 경우 2개의 숫자 생성가능

3. Bottom-Up 방식을 사용해서 해결함

 

※ Dp문제의 경우 각각의 점화식을 찾는게 가장 어렵다