문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
고른 수열은 오름차순이어야 한다.

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

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

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

예제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 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());

        int[] data = new int[m + 1];
        boolean[] picked = new boolean[n + 1];

        picking(n, m, 1, data, picked, 0);
    }

    static void picking(int n, int m, int str, int[] data, boolean[] picked, int cnt) {
        if (cnt == m) {
            for (int i = 0; i < m; i++) {
                System.out.print(data[i] + " ");
            }
            System.out.println();
            return;
        }
        for (int i = str; i <= n; i++) {
            if (!picked[i]) {
                picked[i] = true;
                data[cnt] = i;
                picking(n, m, i + 1, data, picked, cnt+1);
                picked[i] = false; //돌려놓아야지 이후 반복문에서 제외되지 않는다.
            }
        }
    }
}

15649번 문제와 다른 점은 순서와 상관이 없는 수열이라는 점이다.
즉, 1 2 3 4와 4 3 2 1은 같은 것으로 간주한다.
그리고 이때 무조건 모든 배열을 오름차순으로 출력해야되기 때문에 15649번 picked 함수와 달리 str라는 인자를 한 개 더 주어서 이전에 배열에 포함된 수보다 큰 수 중에서만 다음 숫자를 고르도록 했다.

오랜만에 알고리즘 문제를 푸는 거라 간단한 문제임에도 굉장히 당황했던 것 같다.
이제 다시 꾸준히 시작해봐야겠다.