본문 바로가기

Algorithm_BOJ(백준)/동적프로그래밍(DP)

[백준 10844 c++ VV] 쉬운 계단 수

728x90
반응형

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
// [백준 10844 c++ VV] 쉬운 계단 수 
// 문제: n 자리의 계단 수의 개수를 1000000000로 나눈 나머지를 구하는 문제
// 접근1: 직접 갯수 세서 개수간의 규칙으로 점화식 찾아서 찍었는데 틀림
// 점근2: n-1 의 계단수와 n의 계단수의 규칙 찾으려 함 
// 풀이: n자리의 1의 자리가 0인 계단수 : n-1자리의 1의 자리수가 0 또는 2인 계단수의 합
// 식으로 표현 1<= L <=8 dp[n][L] = dp[n-1][L-1]+dp[n-1][L+1]
// n의 1의 자리수가 0 일 때는 1 밖에 안됨 : dp[n][0] =dp[n-1][1]
// n의 1의 자리수가 9 일 때는 8 밖에 안됨 : dp[n][9] =dp[n-1][8]
// 이차원 dp 배열에 n 자리의 L로 끝나는 계단 수의 개수 저장
// 주의: dp값이 커서 long long 자료형 써야 됨
// 주의: 출력에 1000000000 으로 나눈 나머지를 출력해야 하는데 dp 과정중에도 1000000000을 나눈 나머지를 저장해야 답 맞음 -> 이해 안감
// 질문: 출력에 1000000000 으로 나눈 나머지를 출력해야 하는데 dp 과정중에도 1000000000을 나눈 나머지를 저장해야 답 맞음 -> 이해 안감
// 1~9 : 9
// 10 12 21 23 ... 87 89 98 : 17
#define MAX 101
int n;
vector<vector<long long>> dp(MAX,vector<long long>(10));
 
int main()
{    
    ios::sync_with_stdio(false); // 계산시간 단축 // 문제마다 오류 유무 다름
    cin.tie(NULL); cout.tie(NULL);// 입출력 시간 단축
 
    cin >> n;
    // dp 점화식: 
    // N번째 자리 숫자가 0이면 dp[i][j] = (dp[i - 1][j + 1])
    // N번째 자리 숫자가 1~8까지 : dp[N][i] = dp[N-1][i-1] + dp[N-1][i+1]
    // N번째 자리 숫자가 9 일때 : dp[N][i] = dp[N-1][i-1] 
    // 1째 자리 숫자의 계단수 초기값
    for (int i = 0; i <= 9; i++) {
        dp[1][i] = 1;
    }
    for (int i = 2; i <= n; i++) {
        for (int j = 0; j <= 9; j++) {
            if (j == 0) { dp[i][j] = (dp[i - 1][j + 1]) % 1000000000; }
            if (j >= 1 && j <= 8) { dp[i][j] = (dp[i - 1][j - 1+ dp[i - 1][j + 1])% 1000000000; } //N번째 자리 숫자가 1~8까지
            if (j == 9) { dp[i][j] = (dp[i - 1][j - 1])% 1000000000; }//N번째 자리 숫자가 9 일때
        }
    }
    // 정답 출력
    long long answer = 0;
    for (int i = 1; i <= 9; i++) {
        answer += dp[n][i];
    }
    cout << answer % 1000000000 << '\n';
    
    return 0;
}
cs
반응형