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

문제 설명

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]

 

---

C++ 풀이

#include <string>
#include <vector>
#include <cmath>
using namespace std;

vector <int> solution(int n, long long k) {
	vector <int> leftp; //줄을 아직 서지 않은 사람들의 번호
	vector <int> answer;
	vector <long long> fac; 
	fac.push_back(1);
	for (int i = 1; i <= n - 1; i++) //팩토리얼 수 저장
		fac.push_back(i * fac[i - 1]);
	for (int j = 1; j <= n; j++)
		leftp.push_back(j);
	for (int h = 0; h < n; h++) {
		answer.push_back(leftp[ceil(k / (double)fac[n - h - 1]) - 1]);
		leftp.erase(leftp.begin() + ceil(k / (double)fac[n - h - 1]) - 1);
		k %= fac[n - h - 1];
		if (k == 0)
			k = fac[n - h - 1];
	}
	return answer;
}

처음에는 그냥 순열 함수를 써서 단순하게 풀었더니 효율성 테스트를 통과할 수 없었다.

그래서 순열 함수 대신 규칙을 직접 찾아서 다시 풀었다.

 

먼저 사람들의 번호(1 ~ n)를 순차적으로 저장한 벡터 leftp를 생성한다.

 

그 다음 아래 과정을 반복하면서 앞에서부터 차례대로 줄 선 사람의 번호를 구한다.

• i번째로 줄 서있는 사람의 번호는 leftp [  k / (n - i)! 의 올림 값  - 1] 이다.  (*인덱스를 0부터 시작했기 때문에 -1 필요)
• answer 벡터에 i번째 사람 번호 저장,  leftp에서 i번째 사람 번호 삭제
• k값을 k % (n - i)! 값으로 갱신  (* 갱신된 k 값이 0일경우, (n - i)!으로 갱신)

 

 

 

문제 링크

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

코딩테스트 연습 - 줄 서는 방법 | 프로그래머스