백준 15656번 : N과 M (7)

반응형

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

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++){
            if(!check[i]){
                num[depth] = inputNumber[i];
                dfs(n,m, depth+1);
            }
        }
    }


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

DFS를 활용한다. 모든 인덱스의 입력 숫자 만큼 순차적으로 방문해야 하기 때문에 방문 체크 없이 루프를 돌려준다. 

반응형

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

백준 14501번 : 퇴사  (0) 2023.01.11
백준 18290번 : NM과 K (1)  (0) 2023.01.10
백준 15655번 : N과 M (6)  (0) 2023.01.06
백준 15654번 : N과 M (5)  (0) 2023.01.04
백준 15652번 : N과 M (4)  (0) 2023.01.04