‡ CODING TEST STUDY ‡/º 백준
[백준 | Java Silver III] (#15655) N과 M (6)
Trudy | 송연
2024. 8. 5. 00:12
문제
https://www.acmicpc.net/problem/15655
접근
S14544 (N과 M (5)) 문제와 매우 유사했던 문제인데, 오름차순의 경우만 출력해야 했다.
다음과 같이 dfs 코드에서 i=now부터 시작하게 해서, now를 1씩 증가시키면서 dfs가 호출되도록 해서 구현했다.
public static void dfs(int depth, int now){
if(depth == m){
//출력
}
for (int i = now; i < n; i++) {
if (!visited[i]) {
visited[i] = true;
result[depth] = arr[i];
dfs(depth + 1, i+1);
visited[i] = false;
}
}
}
최종코드
package week13.baek;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
public class S15655 {
static int[] arr;
static boolean[] visited;
static int n, m;
static int[] result;
public static void dfs(int depth, int now){
if(depth == m){
for (int i = 0; i < m; i++) {
System.out.print(result[i] + " ");
}
System.out.println();
return;
}
for (int i = now; i < n; i++) {
if (!visited[i]) {
visited[i] = true;
result[depth] = arr[i];
dfs(depth + 1, i+1);
visited[i] = false;
}
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
m = Integer.parseInt(st.nextToken());
st = new StringTokenizer(br.readLine());
arr = new int[n];
for (int i = 0; i <n; i++) {
arr[i] = Integer.parseInt(st.nextToken());
}
Arrays.sort(arr);
visited = new boolean[n];
result = new int[m];
dfs(0, 0);
}
}