티스토리 뷰

www.acmicpc.net/problem/15665

 

15665번: N과 M (11)

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

www.acmicpc.net

 

이 문제는 중복 순열을 구하는 문제입니다. 하나의 숫자는 중복되어 선택할 수 있지만 선택한 수의 조합은 중복이 되지 않도록 해야합니다. 따라서 중복 순열을 구한 뒤, 하나의 문자열로 만들고 그 문자열이 이전에 선택된 수들의 묶음에 포함되지 않는 경우에만 출력합니다.

 

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

 

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

public class Main {
    private static Set<String> answer = new HashSet<>();
    private static StringBuilder answerStr = new StringBuilder();

    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);

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

    public static void permutation(int[] nums, int[] selected, int n, int m, int depth) {
        if (depth == m) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < depth; i++) {
                sb.append(selected[i]).append(" ");
            }

            if (!answer.contains(sb.toString())) {
                answer.add(sb.toString());
                answerStr.append(sb).append("\n");
            }

            return;
        }

        for (int i = 0; i < n; i++) {
            selected[depth] = nums[i];
            permutation(nums, selected, n, m, depth + 1);
        }
    }
}
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/12   »
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30 31
글 보관함