BOJ 15649 N과 M (1)

문제

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

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열

입력

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

출력

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

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

문제 분석

문제 분석
1. 정보
- 자연수 N과 M이 주어졌을 때, 1부터 N까지 자연수 중에서 M개를 고른 수열
2. 목표
- 위 수열을 중복 없이 모두 출력

3. 제약 조건
   • $1 \le M \le N \le 8$

풀이

1. 알고리즘
   • boolean[] check를 통해 해당 숫자의 방문 여부 저장
   • 백트래킹
     • 1 ~ N 부터 시작하는 백트래킹 수행
       • for(1 ~ N)에서 만약 해당 숫자 방문하지 않았다면,
         • 방문처리 하고 배열 해당 순서에 저장
       • 만약 M번 돌았다면 배열 출력 후 return

코드

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

public class BOJ15649 {

    static int N, M;
    static boolean[] check;

    private static void solution() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        check = new boolean[N + 1];
        int[] result = new int[M];
        for (int i = 1; i <= N; i++) {
            check[i] = true;
            result[0] = i;
            recur(result, 1);
            check[i] = false;
        }
    }

    static void recur(int[] result, int cnt) {
        if (cnt == M) {
            for (int cur : result) {
                System.out.print(cur + " ");
            }
            System.out.println();
            return;
        }

        for (int i = 1; i <= N; i++) {
            if (!check[i]) {
                check[i] = true;
                result[cnt] = i;
                recur(result, cnt + 1);
                check[i] = false;
            }
        }
    }

    public static void main(String[] args) throws IOException {
        BOJ15649.solution();
    }
}