백준) 15663. N과 M(9)(자바)
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