취미처럼

[백준] 10972번 다음 순열 본문

Algorithm/백준

[백준] 10972번 다음 순열

sirius 2021. 3. 4. 17:25
https://www.acmicpc.net/problem/10972
 

10972번: 다음 순열

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

www.acmicpc.net

 

  • arr[ i - 1] < arr[ i ] 를 만족하는 가장 큰 i를 찾는다.
  • j >=1 이면서 arr[ j ] > arr[ i - 1] 을 만족하는 가장 큰 j를 찾는다.
  • arr[ i - 1] 과 arr[ j ]를 swap한다.
  • arr[ i ] 부터 순열을 뒤집는다.
import java.util.*;
import java.io.*;

public class Main {

    public static void main(String[] args) throws Exception {
   
        StringBuilder sb = new StringBuilder();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int[] arr = new int[N];

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < N; i++) {
        	arr[i] = Integer.parseInt(st.nextToken());
        }

        if(nextPermutation(arr)) {
            for(int val : arr) {
            	sb.append(val).append(" ");
            }
        } else {
        	sb.append("-1");
        }

        System.out.println(sb);
    }

    public static boolean nextPermutation(int[] arr) {

        int i = arr.length - 1;

        // arr[i-1] < arr[i] 를 만족하는 가장 큰 i
        while(i > 0 && arr[i-1] >= arr[i]) {
        	i--;
        }

        // i의 위치가 0이면 내림차순(마지막 순열)
        if(i<=0) {
        	return false;
        }

        int j = arr.length - 1;

        // j >=i 이면서 arr[j] > arr[i-1] 를 만족하는 가장 큰 i
        while( arr[i-1] >= arr[j]) {
        	j--;
        }

        // arr[i-1] 과 arr[j] 를 swap
        int temp = arr[j];
        arr[j] = arr[i-1];
        arr[i-1] = temp;

        j = arr.length - 1;

        // arr[j] 부터 순열 뒤집기
        while(i < j) {
            temp = arr[i];
            arr[i] = arr[j];
            arr[j] = temp;
            i++;
            j--;
        }

        return true;
    }

}

'Algorithm > 백준' 카테고리의 다른 글

[백준] 10974번 모든 순열  (0) 2021.03.04
[백준] 10973번 이전 순열  (0) 2021.03.04
[백준] 2529번 부등호  (0) 2021.03.04
[백준] 15661번 링크와 스타트  (0) 2021.03.03
[백준] 14889번 스타트와 링크  (0) 2021.03.03
Comments