[백준_15657번] N과 M (8)

https://www.acmicpc.net/problem/15657

15657번: N과 M (8)

[문제]

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다.

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 고른 수열은 비내림차순이어야 한다.

 

[입력 조건]

• 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.입력
• 첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
• 둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

 

[코드]

import java.util.*;

public class BaekJoon_15657 {
	static StringBuilder sb=new StringBuilder();
	static int n,m;
	static int[] num,arr;
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		m=sc.nextInt();
		num=new int[n];
		arr=new int[m];
		
		for(int i=0;i<n;i++)
			num[i]=sc.nextInt();
		
		Arrays.sort(num);
		
		dfs(0,0);
		System.out.println(sb);
	}
	
	static void dfs(int depth, int start) {
		// 선택한 m개의 숫자로 만든 수열 저장
		if(depth==m) {
			for(int i=0;i<arr.length;i++)
				sb.append(arr[i]+" ");
			
			sb.append('\n');
			return;
		}
		
		for(int i=start;i<n;i++) {
			arr[depth]=num[i];
			dfs(depth+1,i);
		}
	}
}

 

[고찰]

 이번 문제는 백트래킹 예제문제 N과 M 8번째 문제이다. N과 M 문제들은 백트래킹을 사용하는 것은 똑같기 때문에 출력해야하는 수열의 조건에 따라 dfs 함수의 매개변수를 달리 해주거나 방문 체크 배열을 써주는 방식으로 해결할 수 있다.