백준 15652: N과 M (4) [JAVA]

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

 

[난이도]

- Silver 3

 

[알고리즘]

- 백트래킹

 

[코드]

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

public class Main {
    static int N, M;
    static boolean[] checked;
    static int[] arr;
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        checked = new boolean[N + 1];
        arr = new int[N + 1];
        backTracking(0, 1);
        bw.flush();
    }

    static void backTracking(int depth, int start) throws Exception {
        if (depth == M) {
            for (int i = 0; i < M; i++) {
                bw.write(arr[i] + " ");
            }
            bw.write("\n");
            return;
        }
        for (int i = start; i <= N; i++) {
                arr[depth] = i;
                backTracking(depth + 1, i);
        }
    }
}

[풀이]

백트래킹 메서드를 실행할 때 매개변수로 depth이외에 추가로 for문을 시작하는 인덱스 start 변수를 준다. 이 start 변수는 i로 주어 재귀함수를 실행한다. 중복을 허용하기 때문에 boolean 배열을 사용하지 않는다.