백준-15649번/N과 M(1) (java)

 

 

문제 접근

 

처음 접하는 백트래킹 문제라서 다소 생소했다.

 

이 문제는 DFS(깊이우선탐색)을 이용해서 수열을 출력해야하는 문제이다.

 

visit배열(boolean) ===> 방문 했는지 안했는지를 판별

arr배열(int) ===> 출력할 숫자를 저장하는 배열

 

 

문제 해결

 

<dfs함수>

private static void dfs(int depth) {
    if (depth == m) {
        for (int val : arr) {
            sb.append(val).append(" ");
        }
        sb.append("\n");
        return;
    }

    for (int i = 0; i < n; i++) {
        // 해당 노드를 방문하지 않았다면
        if (!visit[i]) {
            visit[i] = true; // 해당 노드 방문상태 변경
            arr[depth] = i + 1; // 해당 깊이를 index로 하여 i+1 값 저장
            dfs(depth + 1);// 다음 자식 노드 방문 위해 depth 1 증가시키며 재귀 호출
            visit[i] = false; // 자식노드 방문이 끝나고 돌아오면 방문노드를 방문하지 않은 상태로 변경
        }
    }
}

 

 

N=4  M=2가 입력 되었을경우

 

 

 

<전체 코드>

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

public class Problem15649 {
    static StringBuilder sb = new StringBuilder();
    static int n, m;
    static boolean[] visit;
    static int[] arr;


    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());

        visit = new boolean[n];
        arr = new int[m];
        dfs(0);
        System.out.println(sb);
    }

    private static void dfs(int depth) {
        if (depth == m) {
            for (int val : arr) {
                sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for (int i = 0; i < n; i++) {
            // 해당 노드를 방문하지 않았다면
            if (!visit[i]) {
                visit[i] = true; // 해당 노드 방문상태 변경
                arr[depth] = i + 1; // 해당 깊이를 index로 하여 i+1 값 저장
                dfs(depth + 1);// 다음 자식 노드 방문 위해 depth 1 증가시키며 재귀 호출
                visit[i] = false; // 자식노드 방문이 끝나고 돌아오면 방문노드를 방문하지 않은 상태로 변경
            }
        }
    }
}