[Baekjoon] 다음 순열

👀 문제 설명

문제

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 4 3

 

예제 입력 2

5

5 4 3 2 1

 

예제 출력 2

-1

✍🏻풀이

next_permutation 함수를 사용하면 손쉽게 구할 수 있다.

next_permutation 함수는 순열을 오름차순으로 차례대로 구해주는 함수로, v 벡터에 입력받은 순열을 넣고, next_permutation을 사용하면 자동으로 다음 순열을 구해준다.

 

코드

#include <stdio.h>
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    int N;
    cin >> N;
    
    vector<int> v;
    v.resize(N);
    for (int i = 0; i < N; i++) {
        cin >> v.at(i);
    }
    
    if (next_permutation(v.begin(), v.end())) {
        for (int i = 0; i < N; i++) {
            cout << v.at(i) << " ";
        }
    }
    else {
        cout << -1;
    }
    
    return 0;
}