[백준_15655번] N과 M (6)

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

15655번: N과 M (6)

[문제]

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

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.입력

 

[입력 조건]

• 첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
• 둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

 

[코드]

import java.util.*;

public class BaekJoon_15655 {
	static int n,m;
	static int[] num;
	static boolean[] visited;
	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];
		visited=new boolean[n];
		
		for(int i=0;i<n;i++)
			num[i]=sc.nextInt();
		
		// 수열이 오름차순이도록 정렬한 배열 사용
		Arrays.sort(num);
		
		backtarcking(0,0);
	}

	static void backtarcking(int depth, int start) {
		// 선택한 숫자의 개수가 m개라면, 선택된 숫자들만 출력
		if(depth==m) {
			for(int i=0;i<n;i++) {
				if(visited[i])
					System.out.print(num[i]+" ");
			}
			System.out.println();
			return;
		}
		
		// 백트래킹
		for(int i=start;i<n;i++) {
			if(!visited[i]) {
				visited[i]=true;
				// 중복되는 숫자를 선택하지 않기 위해 재귀호출의 for문 시작을 i+1로 지정
				backtarcking(depth+1,i+1);
				visited[i]=false;
			}
		}
	}
}

 

[고찰]

 이번 문제는 수열을 만들 때 중복되는 숫자를 선택하지 않기 위해 backtracking 함수에 숫자를 선택하는 for문을 제어하는 매개변수를 전달해야 한다. 또한 만들어진 수열이 오름차순이도록 정렬된 배열을 사용해야 한다는 점만 주의하면 쉽게 해결할 수 있는 문제이다.