https://www.acmicpc.net/problem/15655
15655번: N과 M (6)
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열
www.acmicpc.net
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
public class Main{
public static int n, m;
public static int[] arr, nums;
public static boolean[] visit; // 방문 여부
public static StringBuilder sb = new StringBuilder();
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());
arr = new int[m];
visit = new boolean[n+1];
nums=new int[n+1];
st= new StringTokenizer(br.readLine());
for(int i=0; i<n; i++){
nums[i]=Integer.parseInt(st.nextToken());
}
Arrays.sort(nums);
dfs(0, 1);
System.out.println(sb);
}
public static void dfs(int cnt, int start){
if(cnt == m){
for(int val:arr){
sb.append(val).append(" ");
}
sb.append("\n");
return;
}
for(int i=start; i<=n; i++){
if(!visit[i]){
visit[i] = true;
arr[cnt] = nums[i];
dfs(cnt+1, i+1);
visit[i]=false;
}
}
}
}
원소가 중복되지 않도록 방문처리를 해주고, 고른 수열이 오름차순이어야 하므로 for문을 0이 아닌 매개변수로 받은 start 부터 시작하게 한다.
'ALGORITHM' 카테고리의 다른 글
[JAVA] 백준 15656번- N과 M(8) (0) | 2022.09.01 |
---|---|
[JAVA] 백준 15656번- N과 M(7) (0) | 2022.09.01 |
[JAVA] 백준 15654번- N과 M(5) (0) | 2022.09.01 |
[JAVA] 백준 9020번- 골든바흐의 추측 (0) | 2022.08.30 |
[JAVA] 백준 4948번- 베르트랑 공준 (0) | 2022.08.29 |