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

문제 정보

제목 : 줄 서는 방법

난이도 : Lv.2

사용 언어 : Python

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12936

프로그래머스

 

문제 설명

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

입출력 예

n	k	result
3	5	[3,1,2]

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

---

나의 풀이

실패한 풀이

1부터 n+1까지 반복하는 for문을 이용하여 list에 l에 추가한 후, l의 순열을 구하여 저장하고 그 중 k번째에 해당하는 값을 answer에 저장해 리턴했습니다. 이렇게 문제를 해결하니 테스트 케이스는 성공했으나 시간초과로 실패하게 되었습니다.

from itertools import permutations

def solution(n, k):
    answer = []
    l = []
    for n in range(1,n+1) :
        l.append(n)
        
    ls=list(permutations(l,n))
    answer = list(ls[k-1])
    return answer

성공한 풀이

시간을 줄이는 것이 관건이었습니다. 그러기위해선 문제를 다른 방식으로 파악하는 것이 필요했습니다. 각각의 자릿수에 들어가는 숫자는 가장 앞자리가 특정 숫자가 되는 경우는 1~n-1번 방법까지 입니다. 이를 이용하여 한자리씩 맞춰갈 수 있습니다.

먼저 리스트 l에 1부터 n에 해당하는 값을 저장합니다. l에 원소가 들어있는 동안 반복하면서 k를 n-1! 값으로 나눈 정수 몫을 a에 저장합니다. answer에는 l[a]에 해당하는 값을 저장하고 l의 원소는 삭제해줍니다. k는 값을 구한 나머지를 저장하고 n은 -1을 해줍니다. 반복이 끝나면 answer을 리턴합니다.

 

코드

import math

def solution(n, k):
    answer = []
    l = []
    k = k-1
    for n in range(1,n+1) :
        l.append(n)
    
    while l :
        a = k // math.factorial(n-1)
        answer.append(l[a])
        del l[a]
        
        k = k % math.factorial(n-1)
        n -= 1
        
    return answer