백준 10972번 다음 순열 (java)

[Silver III] 다음 순열 - 10972

문제 링크

성능 요약

메모리: 24596 KB, 시간: 408 ms

분류

조합론, 수학

제출 일자

2024년 12월 30일 19:47:05

문제 설명

1부터 N까지의 수로 이루어진 순열이 있다. 이때, 사전순으로 다음에 오는 순열을 구하는 프로그램을 작성하시오.

사전 순으로 가장 앞서는 순열은 오름차순으로 이루어진 순열이고, 가장 마지막에 오는 순열은 내림차순으로 이루어진 순열이다.

N = 3인 경우에 사전순으로 순열을 나열하면 다음과 같다.

• 1, 2, 3
• 1, 3, 2
• 2, 1, 3
• 2, 3, 1
• 3, 1, 2
• 3, 2, 1

예제 입력 1 복사

4
1 2 3 4

예제 출력 1 복사

1 2 4 3

입력

첫째 줄에 N(1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄에 순열이 주어진다.

출력

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

풀이

다음 순열을 찾는 문제다.

순열을 사전순으로 나열한 것을 보니 바로 dfs가 떠올랐다. dfs를 맨 처음부터 완전탐색으로 탐색하면 사전순으로 나열되기 때문이다.

 

하지만 이렇게 완전탐색으로 만들면 시간초과를 벗어날 수 없다.

 

따라서 맨 처음 시작을 입력받은 숫자에서 시작해서 바로 다음 순서의 순열을 찾을 수 있게 dfs를 구성했다.

전체 코드

public class Main {
    static int N;
    static int[] x;
    static int[] sorted;
    static int flag = 0;
    static boolean[] visited;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

       N = Integer.parseInt(br.readLine());
       StringTokenizer st = new StringTokenizer(br.readLine());
       x = new int[N];
       sorted = new int[N];
       visited = new boolean[N];
       for (int i = 0; i < N; i++) {
           x[i] = Integer.parseInt(st.nextToken());
           sorted[i] = i+1;
       }
       dfs(new int[N], 0, x[0]-1);
       System.out.println("-1");
    }

    public static void dfs(int[] re, int depth, int start){
        if(depth == N){
            if(flag == 1){
                for (int i : re) {
                    System.out.print(i + " ");
                }
                exit(0);
            }
            if(check(re) && flag == 0){
                flag = 1;
            }
            return;
        }

        for(int i=start;i<N;i++){
            if(!visited[i]){
                re[depth] = sorted[i];
                visited[i] = true;
                if(depth < N-1 && flag == 0) {
                    dfs(re, depth + 1, x[depth + 1] - 1);
                }
                else{
                    dfs(re, depth + 1, 0);
                }
                visited[i] = false;
            }
        }
    }

    public static boolean check(int[] re){
        for(int i = 0; i < N; i++){
            if(re[i] != x[i]){
                return false;
            }
        }
        return true;
    }
}