시간제한 : 1초
메모리제한 512MB
문제 유형 : 완전탐색
문제 설명 간략 :
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
- N개의 자연수 중에서 M개를 고른 수열
입력
첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.
출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
수열은 사전 순으로 증가하는 순서로 출력해야 한다.
접근법
이전 사용 내역들은 건너 뛰고 사용한 index값은 표시하여 중복을 피한다.
자바 풀이
import java.io.*;
import java.util.*;
public class Main {
static FastReader scan = new FastReader();
static StringBuilder sb = new StringBuilder();
static void input() {
M = scan.nextInt();
N = scan.nextInt();
selected = new int[M + 1];
nums = new int[N + 1];
used = new int[N + 1];
for (int i = 1; i <= N; i++) {
nums[i] = scan.nextInt();
}
Arrays.sort(nums, 1, N + 1);
}
static int N, M;
static int[] selected, nums, last_printed, used;
static void rec_func(int k) {
if (k == M + 1) {
for (int i = 1; i <= M; i++) sb.append(selected[i]).append(' ');
sb.append('\n');
} else {
int last_cand = 0;
for (int cand = 1; cand <= N; cand++) {
if (used[cand] == 1) continue;
if (nums[cand] == last_cand) continue;
last_cand = nums[cand];
selected[k] = nums[cand]; used[cand] = 1;
rec_func(k + 1);
selected[k] = 0; used[cand] = 0;
}
}
}
public static void main(String[] args) {
input();
rec_func(1);
System.out.println(sb.toString());
}
static class FastReader {
BufferedReader br;
StringTokenizer st;
public FastReader() {
br = new BufferedReader(new InputStreamReader(System.in));
}
public FastReader(String s) throws FileNotFoundException {
br = new BufferedReader(new FileReader(new File(s)));
}
String next() {
while (st == null || !st.hasMoreElements()) {
try {
st = new StringTokenizer(br.readLine());
} catch (IOException e) {
e.printStackTrace();
}
}
return st.nextToken();
}
int nextInt() {
return Integer.parseInt(next());
}
long nextLong() {
return Long.parseLong(next());
}
double nextDouble() {
return Double.parseDouble(next());
}
String nextLine() {
String str = "";
try {
str = br.readLine();
} catch (IOException e) {
e.printStackTrace();
}
return str;
}
}
}
출처