[백준] 15654번 - N과 M(5) (Java)

https://www.acmicpc.net/problem/15654

15654번: N과 M (5)

 

<문제해석>

1. 오름차순으로 출력해야 하기 때문에 입력값을 먼저 정렬해야 한다.

2. 숫자 대신 배열 인덱스를 넘겨주고, 배열 A[i]를 저장한다.

3. 중복 조합을 허용하지 않기 때문에 방문배열을 사용한다.

  - 방문한 숫자가 아니면 값을 조합한다.

  - 재귀에서 돌아오면 다음 탐색을 위해 방문처리를 초기화한다.

 

<코드구현 #1>

import java.util.Scanner;
import java.util.Arrays;
import java.lang.StringBuilder;

/*
  N과 M (5)
*/
public class Main {
    static int N, M;
    static int A[];
    static int Ans[];
    static boolean visited[];
    static StringBuilder sb = new StringBuilder();

    void inputData(){
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        M = sc.nextInt();
        A = new int[N];
        Ans = new int[M];
        visited = new boolean[N];
        for (int i=0; i<N; i++) {
            A[i] = sc.nextInt();
        }
        sc.close();
    }

    void DFS(int level) {
        // 종료 조건
        if (level == M) {
            for (int x : Ans) {
                sb.append(x).append(' ');
            }
            sb.append('\n');
            return;
        }

        // 재귀호출 조건
        // 중복 허용 안됨 (방문배열 사용)
        // 순서 바꾸면 다른 조합 인정(i=0부터 시작)
        for (int i=0; i<N; i++) {
            if (!visited[i]) {
                visited[i] = true;
                Ans[level] = A[i];
                DFS(level+1);
                visited[i] = false;
            }
        }
    }
    
    void Solve() {
        Arrays.sort(A);
        DFS(0);
        System.out.println(sb);
    }
 
    public static void main(String[] args) {
        Main m = new Main();
        m.inputData(); // 입력함수
        m.Solve();
    }
}