백준) 10972 다음 순열

10972 다음 순열

문제

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을 출력한다.

예제 입력 

4
1 2 3 4

예제 출력 

1 2 4 3

예제 입력 2 

5
5 4 3 2 1

예제 출력 2 

-1

1. 접근

next_permutation 쓰면 걍 끝날 문제!

전에 풀었던 순열 순서 문제 http://js1jj2sk3.tistory.com/39 보다 더 간단하다.

2. 풀이

한 번 상기시키는 김에 next_permutation의 원리대로 짜보았다.

오른쪽이 전부 내림차순이 아니게 되는 첫 숫자를 찾고, 그 숫자보다 큰 숫자를 뒤에서부터 찾아서 처음 발견되는 숫자와 스위치하면 된다.

나머지는 오름차순으로 정렬해준다.

 

3. 코드

#include <stdio.h>
#include <algorithm>
using namespace std;
 
int n, i, k, arr[10000];
 
int main() {
    scanf("%d", &n);
    for (i = 0; i < n; ++i)
        scanf("%d", &arr[i]);
 
    if (n == 1) {
        printf("-1");
        return 0;
    }
 
    int p = n - 1;
 
    while (arr[p - 1] > arr[p]) {
        p--;
        if (p == 0)
            break;
    }
 
    if (p == 0) {
        printf("-1");
        return 0;
    }
        
    for (k = n - 1; arr[p - 1] > arr[k]; --k);
 
    int tmp = arr[p - 1];
    arr[p - 1] = arr[k];
    arr[k] = tmp;
 
    sort(arr + p, arr + n);
 
    for (i = 0; i < n; ++i)
        printf("%d ", arr[i]);
 
    return 0;
}