[백준_15663번] N과 M (9)

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

15663번: N과 M (9)

[문제]

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

• N개의 자연수 중에서 M개를 고른 수열입력

 

[입력 조건]

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

 

[코드]

import java.util.*;

public class BaekJoon_15663 {
	// 입력 순서를 지키기 위해 HashSet이 아닌 LinkedHashSet을 사용해야 한다. 
	static LinkedHashSet<String> set=new LinkedHashSet<>();
	static int n,m;
	static int[] num,print;
	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();
		
		print=new int[m];
		visited=new boolean[n];
		num=new int[n];
		for(int i=0;i<n;i++)
			num[i]=sc.nextInt();
		
		Arrays.sort(num); // 숫자 정렬
		dfs(0);
		
		// LinkedHashSet에 저장된 수열 출력
		Iterator iter=set.iterator();
		while(iter.hasNext())
			System.out.println(iter.next());
	}
	static void dfs(int depth) {
		// m개의 숫자를 선택했다면 선택한 숫자로 만든 수열 LinkedHashSet에 저장
		if(depth==m) {
			StringBuilder sb=new StringBuilder();
			for(int i:print)
				sb.append(i).append(" ");
			set.add(sb.toString());
			return;
		}
		
		// n개의 숫자 중 m개의 숫자 중복 없이 선택
		for(int i=0;i<n;i++) {
			if(!visited[i]) {
				visited[i]=true;
				print[depth]=num[i];
				dfs(depth+1);
				visited[i]=false;
			}
		}
	}
}

 

[고찰] 

 이번 문제는 백트래킹 기본문제 N과 M 시리즈의 9번째 문제이다. 입력되는 숫자는 중복을 허용하지만 이 수들 중 m개의 숫자를 선택하여 만드는 수열은 중복을 허용하지 않기 때문에 중복을 없애주는 자료구조를 사용하여 해결할 수 있다. 이때 처음에는 HashSet을 사용했는데 HashSet은 입력 순서를 유지하지 않기 때문에 LinkedHashSet을 사용해야한다.