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

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

프로그래머스

풀이

vector의 erase()함수를 사용해 남은 숫자를 편리하게 관리할 수 있습니다.

#include <string>
#include <vector>

using namespace std;

long long fac(int a){
    long long ans = 1;
    for(int i=1;i<=a;i++){
        ans*=i;
    }
    return ans;
}
vector<int> solution(int n, long long k) {
    vector<int> answer;
    vector<int> num;
    long long now=k;
    int cnt=1;
    for(int i=1;i<=20;i++){
        num.push_back(i);
    }
    
    while(cnt != n){
        long long tmp = fac(n-cnt);
        int idx = (now-1) / tmp;
        answer.push_back(num[idx]);
        num.erase(num.begin()+idx);
        cnt++;
        now %= tmp;
        // -1 해줘야 하므로 0일 경우 같은 수로 유지
		if (now == 0)
			now = tmp;
    }
    answer.push_back(num[0]);
    return answer;
    
}