💻코딩테스트 문제풀이24

Chapter12. 구현

[문제55] 줄 서는 방법 - Level2

n명의 사람들이 일렬로 줄을 서고 있습니다. n명의 사람들에게는 각각 1번부터 n번까지 번호가 매겨져 있습니다.

n명의 사람을 줄을 서는 방법은 여러가지 방법이 있습니다. 예를 들어서 3명의 사람이 있다면 다음과 같이 6개의 방법이 있습니다.

• (1, 2, 3)
• (1, 3, 2)
• (2, 1, 3)
• (2, 3, 1)
• (3, 1, 2)
• (3, 2, 1)

사람의 수 n과, 자연수 k가 주어질 때, 사람을 나열하는 방법을 사전 순으로 나열했을 때 k번째 방법을 return 하는 solution 함수를 완성해주세요.

[제한사항]

• n은 20 이하의 자연수입니다.
• k는 n! 이하의 자연수입니다.

[문제풀이]

from itertools import permutations

def solution(n, k):
	return list(permutations([i + 1 for i i range(n)]))[k - 1]

[코드작성]

1. 초기 설정 진행

def solution(n, k):
	numbers = list(range(1, n + 1)
    answer = []
    k -= 1

2. 공식으로 계산하며 숫자 찾기

from math import factorial
while numbers:
	idx, k = divmod(k, factorial(len(numbers) - 1))
    answer.apppend(numbers.pop(idx))

[전체코드]

from math import factorial

def solution(n, k):
	numbers = list(range(1, n + 1)
    answer = []
    k -= 1
    
    while numbers:
		idx, k = divmod(k, factorial(len(numbers) - 1))
    	answer.apppend(numbers.pop(idx))
    
    return answer