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

 

✔️ 입출력 예

입출력 예 #1

문제의 예시와 같습니다.

 

✔️ 코드 구상

문제에 조합을 보자마자 순열이 생각났지만 효율성 테스트가 있는걸 보니 순열로 통과되지 않을 것 같았다.

문제의 예시에 있는 6개 방법의 규칙성을 찾아보니 아래와 같았다.

첫번째 숫자는 n! / n 마다 증가했고 1부터 n까지 늘어 놓았을 때 k / (n! / n) 번째 숫자였다.

우선 List에 1부터 n까지의 숫자를 채우며 n!을 구해준 후 인덱스 0을 고려하기 위해 k-- 를 해준다.

각 자리에 저 규칙을 반복하며 n! / n 을 해줘서 구한 값으로 k / (n! / n)을 해서 채워 넣는다.

 

✔️ 코드

import java.util.*;

class Solution {
    public int[] solution(int n, long k) {
        int[] answer = new int[n];
        List<Integer> list = new ArrayList<>();
        long factorial = 1;
        k--;

        for(int i = 1; i <= n; i++) {
            list.add(i);
            factorial *= i;
        }

        int index = 0;

        while(n > 0) {
            factorial /= n;
            answer[index++] = list.get((int) (k / factorial));
            list.remove((int) (k / factorial));
            k %= factorial;
            n--;
        }

        return answer;
    }
}

 

📄 원문

프로그래머스