백준 15655번 : N과 M (6)

반응형

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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    StringBuilder sb = new StringBuilder();
    int[] num;
    int[] inputNumber;
    boolean[] check;
    public void solution() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringTokenizer stt = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        num = new int[m];
        check = new boolean[n];
        inputNumber = new int[n];
        for(int i = 0; i<n; i++){
            inputNumber[i] = Integer.parseInt(stt.nextToken());
        }
        Arrays.sort(inputNumber);

        dfs(n, m, 0);
        System.out.println(sb);
    }
    public void dfs(int n, int m, int depth){
        if(depth==m){
            for(int a : num){
                sb.append(a).append(" ");
            }
            sb.append("\n");
            return;
        }
        for(int i = 0; i<n; i++){
            for(int j = 0; j<i; j++){
                check[j] = true;
            }
            if(!check[i]){
                check[i] = true;
                num[depth] = inputNumber[i];
                dfs(n,m, depth+1);
                check[i] = false;
            }
            for(int j = 0; j<i; j++){
                check[j] = false;
            }
        }
    }


    public static void main(String args[]) throws IOException {
        new Main().solution();
    }
}

DFS 와 백트래킹을 활용한다. 이미 방문했던 인덱스의 값은 순차적으로 지워가며 방문하기 때문에 j<i만큼 방문 표시해주고, 입력 값을 배열에 담아 정렬해준 후에 재귀한다. 그리고 다시 j<i만큼의 방문표시를 해제하여야 한다. 

반응형

'개발 > 알고리즘' 카테고리의 다른 글

백준 18290번 : NM과 K (1)  (0) 2023.01.10
백준 15656번 : N과 M (7)  (0) 2023.01.06
백준 15654번 : N과 M (5)  (0) 2023.01.04
백준 15652번 : N과 M (4)  (0) 2023.01.04
백준 15651번 : N과 M (3)  (0) 2023.01.04