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

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

1. 문제

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 함수를 완성해주세요.

2. 제한사항

• n은 20이하의 자연수 입니다.
• k는 n! 이하의 자연수 입니다.

3. 풀이

• 고등학교 1학년 팩토리얼에서 배우는 줄 서는 방법 문제다.
• 작년에 고등학생들 가르쳤던 게 도움이 됐다.
• n명의 사람이 줄을 서는 방법은 n!이고 제일 앞 자리 숫자는 (n-1)!개 마다 바뀐다.
• (n-1)! * (n) = n!이니까 당연한건데...
• 그 다음 줄은 (n-2)!개씩 반복된다.
• 그래서 저 (n-i)!로 나눠서 몇 번째 숫자인지 구하면 된다.
• 그런데 몇 번째 숫자라는게 앞에서 몇 번째 숫자를 먼저 선점 했는지에 따라 달라진다.
• 그래서 boolean배열로 확인해서 해줬다.
  나는 설명을 못하는 것 같다.
아 그리고 이 문제는 11점이나 줬다 개꿀이다.

4. 처음 코드와 달라진 점

• K 값을 K % fact(n-1)로 초기화 해주었는데 나누어 떨어졌을 때는 0을 넣으면 안 됐다.

5. 코드

#include <iostream>
#include <vector>

using namespace std;

long long fact(int num){
    long long result = 1;
    for(int i=2; i<=num; i++)
        result*=i;
    return result;
}

vector<int> solution(int n, long long k) {
    vector<int> answer;
    vector<int> orderArr;
    
    for(int i = 1; i < n; i++){
        long long factNum = fact(n-i);
        int order = (k-1) / factNum;
        orderArr.push_back(order+1);
        
        k =  k % factNum==0 ? factNum : k % factNum;
    }
    
    vector<bool> check(n+1);
    for(int order : orderArr){
        int count=0;
        for(int i = 1; i <= n; i++){
            if(!check[i])    count++;
            if(count==order) {
                check[i] = true;
                answer.push_back(i);
                break;
            }
        }    
    }
    
    for(int i = n; i >= 1; i--){
        if(!check[i]) answer.push_back(i);
    }
    
    return answer;
}