숨막히는 알고말고/문제 풀이
[Baekjoon] 이전 순열
숨숨숨
2021. 3. 14. 02:35
👀 문제 설명
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을 출력한다.
예제 입력 1
4
1 2 3 4
예제 출력 1
-1
예제 입력 2
5
5 4 3 2 1
예제 출력 2
5 4 3 1 2
✍🏻풀이
이전에 푼 BOJ 10972 다음 순열과 같은 유형의 문제이다.
이전 순열을 구하려면, next_permutation을 사용해 간단히 구해줄 수 있다.
next_permutation의 세번째 인자인 compare 함수를 사용하면 된다. compare 함수에 greater<int>()를 입력하면, 내림차순으로 순열을 구한다는 뜻으로, 입력하지 않을 경우에는 기본값인 less<int>() 즉, 오름차순으로 순열을 구하게 된다.
코드
#include <stdio.h>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
ios::sync_with_stdio(0);
cin.tie(NULL); cout.tie(NULL);
int N;
cin >> N;
vector<int> v;
for (int i = 0; i < N; i++) {
int num;
cin >> num;
v.push_back(num);
}
if (next_permutation(v.begin(), v.end(), greater<int>())) {
for (int i = 0; i < v.size(); i++) {
cout << v.at(i) << " ";
}
}
else {
cout << -1;
}
return 0;
}