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

🔗 문제 링크

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

📂 분류

구현 수학

💡 풀이

이 문제는 규칙만 찾는다면 쉽게 해결할 수 있는 문제이다.

 

접근 방식

1. n!를 구한다.
2. n! / n , k % (n! / n)를 구한다. 여기서 몫이 0이라면 -1 해준다.
3. 2번의 해당하는 요소를 answer에 넣는다. n은 -1 하고, k는 num으로 나눈 나머지를 넣어준다.
4. 1, 2, 3번 과정을 n이 0이 될 때까지 반복한다.

 

💻 코드

from math import factorial


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

    number = [i for i in range(1, n + 1)]

    while not n == 0:
        num = factorial(n) // n
        idx = k // num if k % num else k // num - 1

        answer.append(number.pop(idx))

        n, k = n - 1, k % num

    return answer