N과 M (9) (15663번)
https://www.acmicpc.net/problem/15663
풀이방법
- 중복된 숫자가 주어질 수 있지만 중복된 순열은 출력하면 안된다는 조건이 있으므로
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
55
56
57
58
59
60
61
62
import java.io.IOException;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Iterator;
import java.util.StringTokenizer;
import java.util.LinkedHashSet;
import java.util.Arrays;
import java.util.Set;
public class Main {
private static Set<String> set = new LinkedHashSet<>();
private static StringBuilder sb = new StringBuilder();
private static int[] result = null;
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];
boolean[] visited = new boolean[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, visited, M, 0);
sb.delete(0, sb.length());
Iterator<String> iter = set.iterator();
while (iter.hasNext()) {
sb.append(iter.next()).append('\n');
}
System.out.println(sb);
}
private static void solve(int[] arr, boolean[] visited, int r, int c) {
if (c == r) {
for(int i : result) {
sb.append(i).append(' ');
}
set.add(sb.toString());
sb.delete(0, sb.length());
return;
}
else {
for(int i=0; i<arr.length; i++) {
if (!visited[i]) {
visited[i] = true;
result[c] = arr[i];
solve(arr, visited, r, c+1);
visited[i] = false;
}
}
}
}
}
Comments powered by Disqus.