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

문제주소 :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
문제의 예시와 같습니다.

 

<풀이법>

▒ 한줄 개념: 인덱싱 ▒ 

많은 분들이 쉽게 풀었다고 하시던데, 저한텐 꽤나 까다로운 인덱싱 문제였습니다.

수학적으로 접근해야하는 문제여서 그런가, 좀 까다로웠던 것 같네요.

풀이 방법은, k-1을 (n-1)! 으로 나눠준 인덱스의 숫자를 answer에 차곡차곡 쌓는 것입니다.

n = 3, k = 5라고 할때, numbers = [1,2,3] 로 초기화해주고, 미리 k와 n을 각각 1씩 빼줍니다.

 

k -= 1

n -= 1

numbers[ k // n! ] = numbers [ 4 // 2! ] = numbers [ 2 ] -> numbers.pop(2) = 3 -> answer.append(3)

 

그리고 k는 앞 연산의 나머지가 되고, n은 1을 빼주면서 반복을 진행합니다.

 

k = k % n 

n = n - 1

numbers[ k // n! ] = numbers [ 0 // 1! ] = numbers [ 0 ] -> numbers.pop(0) = 1 -> answer.append(1)

 

 

<코드(Python)>

def solution(n, k):
    answer = []
    numbers = [i+1 for i in range(n)]
    total = [1]
    for i in range(1,n+1):
        total.append(total[i-1] * i)
    
    n -= 1
    k -= 1
    while len(numbers) != 1:
        index = k // total[n] 
        answer.append(numbers[index])
        numbers.pop(index)
        k = k % total[n]
        n -= 1
    
    answer += numbers
    
    return answer

 

 

더 많은 코드 보기(GitHub) : github.com/dwkim-97/CodingTest