N과 M (10) (15664번)
https://www.acmicpc.net/problem/15664
풀이방법
- 중복된 숫자가 주어질 수 있지만 중복된 순열은 출력하면 안된다는 조건이 있으므로
LinkedHashSet을 이용하였다. - 일반적인
HashSet은 순서가 보장되지 않으나LinkedHashSet은 입력된 순서에 따라 순서가 보장된다. - 먼저 입력받은 숫자들을 정렬한 후
solve함수에 전달해주었다. solve함수의 전달인자는 다음과 같다. 1) 첫번째 : 데이터가 들어있는 배열 2) 두번째 : 현재 데이터 배열에 접근한 인덱스가 몇인지 전달하여 그 이후의 데이터들로만 고르도록 하기 위한 변수 3) 세번째 : 몇 개를 고를 것인지에 대한 변수 4) 네번째 : 현재 뽑은 개수 및 데이터 저장 위치
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import java.io.IOException;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
import java.util.Arrays;
import java.util.Iterator;
import java.util.LinkedHashSet;
public class Main {
private static StringBuilder sb = new StringBuilder();
private static int[] result = null;
private static LinkedHashSet<String> ls = new LinkedHashSet<>();
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());
int[] arr = new int[N];
result = new int[M];
st = new StringTokenizer(br.readLine());
int index = 0;
while (st.hasMoreTokens()) {
arr[index++] = Integer.parseInt(st.nextToken());
}
Arrays.sort(arr);
solve(arr, 0, M, 0);
Iterator<String> iter = ls.iterator();
while (iter.hasNext()) {
sb.append(iter.next()).append('\n');
}
System.out.println(sb);
}
private static void solve(int[] arr, int curIndex, int r, int c) {
if (r == c) {
for(int i : result) {
sb.append(i).append(' ');
}
ls.add(sb.toString());
sb.delete(0, sb.length());
}
else {
for (int i=curIndex; i<arr.length; i++) {
result[c] = arr[i];
solve(arr, i+1, r, c+1);
}
}
}
}
Comments powered by Disqus.