[BOJ] [Python] 1038번: 감소하는 수

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

1038번: 감소하는 수

문제

---

음이 아닌 정수 X의 자릿수가 가장 큰 자릿수부터 작은 자릿수까지 감소한다면, 그 수를 감소하는 수라고 한다. 예를 들어, 321과 950은 감소하는 수지만, 322와 958은 아니다. N번째 감소하는 수를 출력하는 프로그램을 작성하시오. 0은 0번째 감소하는 수이고, 1은 1번째 감소하는 수이다. 만약 N번째 감소하는 수가 없다면 -1을 출력한다.

 

입력

---

첫째 줄에 N이 주어진다. N은 1,000,000보다 작거나 같은 자연수 또는 0이다.

 

 

출력

---

첫째 줄에 N번째 감소하는 수를 출력한다

 

풀이방법

---

조합(Combination)을 이용해 문제를 해결하였다.

1 자릿수 >> 10 Combination 1 => 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 

2 자릿수 >> 10 Combination 2 => (0,1), (0,2), (0,3) ... (8,9)

* 여기서 tuple을 뒤집어서 합친다면 (10, 20, 30, ... 98 ) 이러한 방식으로 나온다. 

이러한 방식으로 10 Combination 1 to 10 까지 배열에 추가한 뒤, 배열을 정렬하면 감소하는 수의 배열이 된다.

배열의 길이 < N + 1 인 경우 : -1 출력, 그 외의 경우 배열[N]을 출력한다.

 

코드

---

from itertools import combinations

ans = []
comb = list(range(10))

for i in range(1,11):
    for j in combinations(comb,i):
        ans.append(int(''.join(list(map(str,reversed((j)))))))

ans.sort()

n = int(input())

if len(ans) < n+1:
    print(-1)
else:
    print(ans[n])