알고리즘 - LV3.줄 서는 방법

문제

programmers.co.kr/learn/courses/30/lessons/12936

코딩테스트 연습 - 줄 서는 방법

 

코드

1) itertools모듈의 permutations이용 => 경우의 수를 불필요하게 많이 계산하게 됨

import itertools

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

 

2) math모듈의 factorial이용

import math

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

    while n != 0:
        facto = math.factorial(n - 1)
        answer.append(numbers.pop((k - 1) // facto))
        n -= 1
        k %= facto

    return answer