감소하는 수_1038

문제

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

입력

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

출력

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

예제 1

예제 2,3

코드

from itertools import combinations

N = int(input())

result = []
for i in range(1,11):
    for j in combinations(range(10), i): 
        num = ''.join(list(map(str, reversed(list(j)))))
        result.append(int(num))

result.sort()
if N >= len(result):
   print(-1)
else:
    print(result[N])

문제 접근

1. python의 combination함수를 이용해 수의 조합을 뽑아내기로 결정
2. for i in range(1,11):
   for j in combinations(range(10), i):
   • 현재 자릿수에 대한 가능한 조합을 생성합니다. 예를 들어, 2자리 수인 경우 (0부터 9까지 의 숫자 중에서) 가능한 모든 2자리 조합을 생성
3. num = ''.join(list(map(str, reversed(list(j)))))): 생성된 조합을 역순으로 정렬하여 감소하는 수를 만듭니다. 숫자를 문자열로 변환하고 역순으로 배열하여 감소하는 숫자를 생성합니다.
4. 생성된 감소하는 수를 정수로 반환하여 리스트로 저장됨
   
   이제 여기까지 하면 출력된다.
5. 이제 입력값에 해당하는 것을 리스트의 길이를 통해 뽑아내면 된다.
   이때, N이 결과 리스트의 길이보다 크거나 같다면, N번째 감소하는 수는 존재하지 않으므로 -1을 출력하는 예외처리만 해주면 끝.
   ex) print(len(result))의 결과 1023
   1022번째 감소하는 수
   
   1023번째 감소하는 수
   print(result[1023])에 대한 결과
   

이제 잘 예외처리가 된 것을 확인하였다.

느낀 점

맨날 자바만 하다가 이번에 처음 파이썬으로 코테를 시작해보았는데 왜 파이썬을 다들 쓰는지 알 것 같았다. combination 객체를 처음 알게되었고 이것을 잘 활용하면 간결한 코드를 짤 수 있을 듯 하다. 처음 골드문제를 푸는 것이라 이번에 구글의 도움을 얻었지만 최대한 코드를 이해하고 나의 것으로 학습시키려고 노력했다. 특히 조합을 구현해야하는 알고리즘 문제를 풀면서 for문을 중첩해서 풀기 보단 itertools를 활용하여 간단하게 구현하는 것을 알게되었다.