숨막히는 알고말고/문제 풀이

[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;
}