백준 15654: N과 M (5) [JAVA]

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

 

[난이도]

- Silver 3

 

[알고리즘]

- 백트래킹

 

[코드]

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

public class Main {
    static int N, M;
    static boolean[] checked;
    static int[] arr, tmp;
    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];
        arr = new int[N];
        tmp = new int[N];

        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(arr); // 오름차순 정렬
        backTracking(0);
        bw.flush();
    }

    static void backTracking(int depth) throws Exception {
        if (depth == M) { // depth 가 M이 될때까지 실행
            for (int i = 0; i < M; i++) {
                bw.write(tmp[i] + " ");
            }
            bw.write("\n");
            return;
        }
        for (int i = 0; i < N; i++) {
            if (!checked[i]) { // 사용한 숫자 체크
                checked[i] = true;
                tmp[depth] = arr[i];
                backTracking(depth + 1);
                checked[i] = false; // 완전탐색을 위해 사용한 숫자는 false
            }

        }
    }
}

[풀이]

지금까지 풀어왔던 N과 M 문제들과 다를 바 없다. 다른점은 Arrays.sort(int[])로 오름차순 정렬을 해주어야 한다.