[백준] 15664번 - N과 M(10) (Java)

 

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

15664번: N과 M (10)

 

<문제해석>

1. 입력받은 정수배열만 사용해야 함.

2. 중복수열 허용안함 (방문배열 사용함)

3. 순서 바뀌면 다른 조합으로 인정안함(i = start 부터 시작)

4. 입력 배열 중 같은 숫자가 있으면 중복 방문은 아니지만 같은 조합이므로 한번만 출력해야 함.

  - HashSet을 사용해서 중복 문자열이 있으면 머지하지 않고 스킵하도록 구현함.

 

<코드구현 #1>

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

public class Main {
    static int N, M;
    static int A[];
    static int Ans[];
    static boolean visited[];
    static StringBuilder sb = new StringBuilder();
    static HashSet<String> set = new HashSet<>();

    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 start, int level) {
        // 종료 조건
        if (level == M) {
            StringBuilder sb2 = new StringBuilder();
            for (int x : Ans) {
                sb2.append(x).append(" ");
            }
            // 기존 머지된 출력물에 동일 문자열이 있는지 확인
            if (!set.contains(sb2.toString())) { // 없으면 머지함
                sb.append(sb2.toString()).append("\n");
                set.add(sb2.toString());
            }
            return;
        }

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