백준 15650번 : N과 M (2)

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

 

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

public class Main {
    StringBuilder sb = new StringBuilder();
    int[] num;
    boolean[] check;
    public void solution() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        num = new int[m];
        check = new boolean[n];
        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]){
                for(int j = 0; j<i; j++){
                    check[j] = true;
                }
                check[i] = true;
                num[depth] = i+1;
                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와 백트래킹을 활용한다. 예제 출력2를 보면 기존에 유효체크 했던 값은 다음 루프에서 더 이상 방문하지 않는다.

현재 인덱스 이하는 전부 체크(방문확인) 해주고, 나올때 다시 체크 해제 해준다.