줄 서는 방법

줄 서는 방법

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 함수를 완성해주세요.

---

풀이

효율성 테스트 불합

from itertools import permutations
def solution(n, k):
    answer = []
    arr = []
    for i in range(1,n+1):
        arr.append(i)
    for i in permutations(arr,n):
        answer.append(list(i))

    for i in range(len(answer)):
        if i+1 == k:
            return answer[i]

코드 리팩토링

def factorial(n):
  if n < 1:
      return 1
  else:
      return n * factorial(n - 1)

def solution(n, k):
  # 총 n!의 경우의 수를 갖고있음
  # 맨 앞의 수가 정해지면 (n - 1)! 의 경우의 수를 가짐
  # k에서 (n - 1)!을 나눴을 때의 몫이 첫번째 오는 자리
  # k에서 (n - 1)!을 나눴을 때의 나머지가 다시 k로 오는 자리
  result = []
  num_list = [i for i in range(1, n + 1)]
  print(num_list)
  while(n != 0):
      num_case = factorial(n - 1)
      print(num_case)
      idx = k // num_case #현재 자리에 올 숫자의 인덱스
      k = k % num_case # 다음 자리를 구할 때 사용
      print("몫 나머지", idx,k)
      if k == 0:
          result.append(num_list.pop(idx - 1))
          print("result", result)
      else:
          result.append(num_list.pop(idx))
          print("result", result)
      n -= 1
  return result

from math import factorial
def solution(n, k):
 answer = []
 nums = list(range(1, n+1))
 #print(nums)
 order = []
 k -= 1
 for i in range(n-1, -1, -1):
     temp = factorial(i)
     print(i,k,temp)
     q, r = divmod(k, temp)
     k = r
     order.append(q)
 # print(order)
 for idx in order:
     answer.append(nums.pop(idx))

 return answer