문제 출처
https://www.acmicpc.net/problem/15652
풀이
- 백트래킹 문제이다.
- N과 M (3)의 풀이에서 dfs에 넘길 index만 수정해주면 쉽게 풀 수 있다.
- 시간 초과가 날 수 있으므로 BufferdWriter를 사용한다.
코드
import java.io.*;
import java.util.StringTokenizer;
public class Main {
static BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(System.out));
public static void main(String[] args) throws IOException {
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
int N = Integer.parseInt(stringTokenizer.nextToken());
int M = Integer.parseInt(stringTokenizer.nextToken());
bufferedReader.close();
int[] arr = new int[N];
dfs(N, M, arr, 1, 0);
bufferedWriter.flush();
bufferedWriter.close();
}
private static void dfs(int N, int M, int[] arr, int index, int depth) throws IOException {
if (depth == M) {
for (int i = 0; i < M; i++) {
bufferedWriter.write(String.valueOf(arr[i]) + ' ');
}
bufferedWriter.newLine();
return;
}
for (int i = index; i <= N; i++) {
arr[depth] = i;
dfs(N, M, arr, i, depth + 1);
}
}
}
채점 결과