원래 DP 문제였는데... 내가 푼 게 dp가 아닌 것 같다...
백준 10844번. 쉬운 계단 수
1. 문제
45656이란 수를 보자.
이 수는 인접한 모든 자리수의 차이가 1이 난다. 이런 수를 계단 수라고 한다.
세준이는 수의 길이가 N인 계단 수가 몇 개 있는지 궁금해졌다.
N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구하는 프로그램을 작성하시오. (0으로 시작하는 수는 없다.)
2. 입력
- 첫째 줄에 N이 주어진다. N은 1보다 크거나 같고, 100보다 작거나 같은 자연수이다.
3. 출력
- 첫째 줄에 정답을 1,000,000,000으로 나눈 나머지를 출력한다.
4. 풀이
- 1자리 수는 모두 쉬운 계단 수로 처리한다.
- 이것저것 계산 해보았다.
- 2자리 수는 10, 12, 23, 34, 45, 56, 67 , 78, 89, 21, 32, .... 등등이 있다.
- 내가 생각한 건 계단 수에서 마지막 숫자의 +1/-1 한 숫자를 붙이면 쉬운 계단 수가 된다.
- 근데 0은 -1을 못하고 9는 +1을 못 한다.
- 배열 한 개로는 안 되고 두 개를 번갈아가면서 썼다.
5. 처음 코드와 달라진 점
- 숫자가 커져서 나머지 계산을 하는게 좀 틀렸다.
- 코드 자체는 바뀌지 않았는데 데이터 타입을 바꿔주거나 나머지 연산을 여기저기 붙이니까 해결됐다.
- 기본 로직은 바뀌지 않았는데 나머지 연산 붙이면서 좀 지저분해졌다.
6. 코드
#include <iostream>
#include <algorithm>
using namespace std;
long long d[10], dTmp[10];
int main() {
int n;
cin >> n;
for (int i = 1; i < 10; i++){
d[i] = 1;
}
for (int i = 1; i < n; i++) {
for (int j = 0; j < 10; j++) {
dTmp[j] = d[j];
}
d[0] = dTmp[1];
d[9] = dTmp[8];
for (int j = 1; j < 9; j++) {
d[j] = (dTmp[j-1] + dTmp[j+1]) % 1000000000;
}
}
long long answer = 0;
for (int i = 0; i < 10; i++) {
answer += (d[i] % 1000000000);
}
answer %= 1000000000;
cout << answer << endl;
}