[백준알고리즘] 15656 - N과 M (7)

www.acmicpc.net/problem/15656

15656번: N과 M (7)

 

문제는 위와 같으며 주어진 숫자를 가지고 중복 순열을 구하는 문제입니다. 중복 순열이기 때문에 해당 수를 선택했는지 아닌지 확인하지 않고 바로 처음부터 수를 확인하며 선택하고 선택한 수 배열에 추가합니다.

 

자바 코드는 다음과 같습니다.

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");

        int n = Integer.parseInt(input[0]);
        int m = Integer.parseInt(input[1]);

        int[] nums = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        Arrays.sort(nums);

        System.out.println(permutation(nums, new int[n], n, m, 0));
    }

    public static StringBuilder permutation(int[] nums, int[] selected, int n, int m, int depth) {
        StringBuilder sb = new StringBuilder();

        if (depth == m) {
            for (int i = 0; i < depth; i++) {
                sb.append(selected[i]).append(" ");
            }

            return sb.append("\n");
        }

        for (int i = 0; i < n; i++) {
            selected[depth] = nums[i];
            sb.append(permutation(nums, selected, n, m, depth + 1));
        }

        return sb;
    }
}