1 minute read

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

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

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

입출력1

입력 : 3 1
4 4 2

출력 :
2
4

입출력2

입력 : 4 2
9 7 9 1

출력 :
1 7
1 9
7 1
7 9
9 1
9 7
9 9

= 4개중 2개를 뽑는다(9 7 9 1 중 2개를 뽑아서 사전 순 출력)

  • 중복 불가
  • N개 중에 M개 뽑기
  • 사전 순

(1) 같은 숫자를 여러번 뽑으면 안되므로 visited 배열로 다녀간 부분 체크
(2) 뽑은 숫자들이 중복되면 안되므로 Set 사용
(TreeSet을 썼더니 실패 -> LinkedHashSet 사용)
(3) 사전 순 출력하기 위해 미리 정렬

import java.util.*;

public class Main {

         static int N, M;
         static int[] selected, numArr;
         static boolean[] visited; // (1)
         static LinkedHashSet<String> set = new LinkedHashSet<>(); // (2)

        public static void main(String[] args){
            input();
            recursion(1);
            for(String s : set) {
                System.out.println(s);
            }
        }
        public static void input(){
            Scanner sc = new Scanner(System.in);
            N = sc.nextInt();
            M = sc.nextInt();
            selected = new int[M+1];
            visited = new boolean[N];
            numArr = new int[N];
            for(int i=0; i<numArr.length; i++) {
                numArr[i] = sc.nextInt();
            }
            Arrays.sort(numArr); // (3)
        }

        public static void recursion(int k) { 
            if(k == M+1) {
                StringBuilder sb = new StringBuilder();
                for(int i=1; i<=M; i++) {
                    sb.append(selected[i]).append(' ');
                }
                set.add(sb.toString());
            } else {
                for(int cand=0; cand<N; cand++) {
                    if(visited[cand]) {
                        continue;
                    }
                    visited[cand] = true;
                    selected[k] = numArr[cand];
                    recursion(k+1);
                    visited[cand] = false;
                }
            }
        }
        
} 

참고 : 한 번에 끝내는 코딩테스트 369 Java편 초격차 패키지 Online part3