Baekjoonアルゴリズム | 15650番問題: NとM (2)

Tags:

出典

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

問題

Baekjoon Online Judgeの15650番問題、**NとM (2)**を解きます。

入力

正確な入力形式と制約は元の問題文に従います。

出力

問題で求められる答えを出力します。

サンプル入力 1

3 1

サンプル出力 1

1 2 3

サンプル入力 2

4 2

サンプル出力 2

1 2 1 3 1 4 2 3 2 4 3 4

サンプル入力 3

4 4

サンプル出力 3

1 2 3 4

アルゴリズム分類

  • バックトラッキング

解説

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

public class Main {

    static StringBuilder sb = new StringBuilder();
    static int n, m;
    static int[] arr;

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

        arr = new int[m];
        dfs(0, 1);
        System.out.println(sb);
    }

    /**
     * 深さ優先探索 (DFS)
     */
    static void dfs(int depth, int start) {
        if (depth == m) {
            for (int value : arr) {
                sb.append(value).append(" ");
            }
            sb.append("\n");
            return;
        }
        for (int i = start; i <= n; i++) {
            arr[depth] = i;
            dfs(depth + 1, i + 1);
        }
    }
}