CodingTest/백준

[백준] 백트래킹 (15649 : N과 M (1))

브디크리 2023. 9. 13. 22:37

문제


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

 

15649번: N과 M (1)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net


해설


import java.io.*;
import java.util.StringTokenizer;

public class Main {
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    static StringBuilder sb = new StringBuilder();

    static int arr[];
    static boolean visit[];
    public static void main(String[] args) throws IOException {
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        arr = new int[M];
        visit = new boolean[N];

        dfs15649(N, M, 0);

        bw.write(sb.toString());
        bw.flush();
        bw.close();
    }

    private static void dfs15649(int N, int M, int depth){
        if(depth == M){
            for (int i : arr)
                sb.append(i).append(" ");
            sb.append("\n");
            return;
        }
		
        for (int i = 0; i < N; i++) {
            if(!visit[i]){
                visit[i] = true;
                arr[depth] = i+1;
                dfs15649(N, M, depth+1);
                visit[i] = false;
            }
        }
    }
}

  1. 재귀함수에서 M == depth일 때, 재귀함수를 탈출하고 출력문을 추가하도록 한다.

  2. 중복되는 수 ({1, 1}, {2,2} 등)가 나오면 안되기 때문에 visit이라는 boolean 배열을 사용하여,
    visit이 false일 경우에만 배열에 추가해주었다.

  3. 재귀함수를 호출하고 탈출한 이후, visit을 false로 다시 변경해주어, 
    다음 for문에서 해당 수를 사용할 때 지장이 없도록 만들어주어야 한다.

결론


백트래킹 문제를 처음 접해보다 보니, 어떤 식으로 풀어야할지 막막하게 느껴졌지만, 

새로운 것을 알게된다는 것이 재미있기도 했다!!.