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

 

문제

https://www.acmicpc.net/problem/10844

 

45656이란 수를 보자.

이 수는 인접한 모든 자리의 차이가 1이다. 이런 수를 계단 수라고 한다.

N이 주어질 때, 길이가 N인 계단 수가 총 몇 개 있는지 구해보자. 0으로 시작하는 수는 계단수가 아니다.

 

 

 

입력

 

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

 

 

출력

 

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

 

 

풀이

 

계단수의 특성을 생각하면 쉬운 문제다. 계단수는 연속적이여야 하므로 N번째 자리에 1이라는 수가 오기 위해서는 N-1번째 자리에는 0또는 2가 와야만 한다.

 

그러므로 N번째 자리가 1인 계단수는 N-1번째 자리가 0인 수와 2인 수의 합과 갯수가 같다. 이를 코드로 작성해 문제를 풀었다.

 

 

Code

 

 

lst=[1]*10
lst[0]=0
ans=[]
ans=lst[::]
N=int(input())
for i in range(N-1):
    for j in range(10):
        if j==0:
            ans[0]=lst[1]
        elif j==9:
            ans[9]=lst[8]
        else:
            ans[j]=lst[j-1]+lst[j+1]
    lst=ans[::]
print(sum(ans)%1000000000)

 

---

 

 

 

정말 쉬운 계단 수 문제였다.

 

​