백준 10972번 다음 순열 // C++

문제

 

1부터 N까지의 수로 이루어진 순열이 있다. 이때, 사전순으로 다음에 오는 순열을 구하는 프로그램을 작성하시오.

사전 순으로 가장 앞서는 순열은 오름차순으로 이루어진 순열이고, 가장 마지막에 오는 순열은 내림차순으로 이루어진 순열이다.

N = 3인 경우에 사전순으로 순열을 나열하면 다음과 같다.

• 1, 2, 3
• 1, 3, 2
• 2, 1, 3
• 2, 3, 1
• 3, 1, 2
• 3, 2, 1

입력

첫째 줄에 N(1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄에 순열이 주어진다.

출력

첫째 줄에 입력으로 주어진 순열의 다음에 오는 순열을 출력한다. 만약, 사전순으로 마지막에 오는 순열인 경우에는 -1을 출력한다.

 

---

 

풀이

 

next_permutation을 이용하여, 다음 순열이 존재한다면 해당 순열을 존재하지 않는다면 -1을 출력하면 된다.

 

다음 순열이 존재할 경우, next_permutation 함수는 true를 리턴하기 때문에 이 경우에만 출력해주면 된다.

 

---

 

코드

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

int main() {
	int n;
	cin >> n;

	vector<int> v;
	for (int i = 0; i < n; i++) {
		int input;
		cin >> input;

		v.push_back(input);
	}

	if (next_permutation(v.begin(), v.end())) {
		for (int i = 0; i < n; i++) {
			printf("%d ", v[i]);
		}
	}
	else
		printf("-1\n");

	return 0;
}