[프로그래머스] 줄 서는 방법

문제

문제 설명

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! 이하의 자연수 입니다.

 

입력 & 출력

입출력 예

nkresult

3

5

[3,1,2]

 

풀이 과정

코드2번의 방식으로 수열을 이용한 풀이를 생각하였지만 효율성이 전부 문제가 생기면서 실패가 되었습니다.

하여 코드1번의 방식으로 해결하였습니다.

1. 코드1번의 방법은 도저히 떠오르지 않아서 검색을 통하여 알아보았습니다. 기본적으로 팩토리얼을 이용한 방법입니다.

2. k를 (n-1)!로 나누면 k번째 방법ㅂ에 어떤 수가 가장 앞에 있는지 알수 있는 것을 이용합니다.

3. 파이썬 내장 함수인 factorial을 이용하며 n이 점점 감소할 수 있도록 while문을 구성하고 n이 0이되면 탈출 할 수 있게 구성하여 answer를 구성하고 리턴하도록 합니다.

 

코드

코드 1

from math import factorial


def solution(n, k):
    answer = []
    order = list(range(1, n + 1))
    while n != 0:
        fact = factorial(n - 1)
        answer.append(order.pop((k - 1) // fact))
        n = n - 1
        k = k % fact
    return answer

 

코드 2

import itertools


def solution(n, k):
    answer = []
    person = []

    for i in range(1, n + 1):
        person.append(i)

    arr = itertools.permutations(person, n)

    for i, ar in enumerate(arr):
        if i == k:
            for a in ar:
                answer.append(a)

    return answer