[Python/BOJ] 1038. 감소하는 수

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

문제

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

입력

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

출력

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

풀이

Code

import sys
from itertools import combinations
input = sys.stdin.readline

# 1. 출력 리스트 생성
ans = []
# 2.
for i in range(1, 11) :
    # 2-1.
    for combination in combinations(range(10), i) :
        # 2-1-1. 조합 내림차순 정렬 후 정수로 변환
        num = int(''.join([str(s) for s in sorted(combination, reverse = True)]))
        # 2-1-2. 출력 리스트에 삽입
        ans.append(num)
# 3. 출력 리스트 정렬
ans.sort()
# 4. 결과 출력
n = int(input())
print(ans[n] if n < len(ans) else -1)