[백준_15650] N과 M (2) - JAVA

문제 출처

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

풀이

• 백트래킹 문제이다.
• index와 depth를 증가시키며, 중복된 순열이 나오지 않도록 한다.

코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        int N = Integer.parseInt(stringTokenizer.nextToken());
        int M = Integer.parseInt(stringTokenizer.nextToken());
        bufferedReader.close();
        int[] arr = new int[N];

        dfs(N, M, arr, 1, 0);
    }

    private static void dfs(int N, int M, int[] arr, int index, int depth) {
        if (depth == M) {
            StringBuilder stringBuilder = new StringBuilder();
            for (int i = 0; i < M; i++) {
                stringBuilder.append(arr[i]).append(" ");
            }
            System.out.println(stringBuilder);
            return;
        }

        for (int i = index; i <= N; i++) {
            arr[depth] = i;
            dfs(N, M, arr, i + 1, depth + 1);
        }
    }

}

채점 결과