[백준_15665번] N과 M (11)

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

15665번: N과 M (11)

[문제]

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

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.입력

 

[입력 조건]

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

 

[코드]

import java.util.*;

public class BaekJoon_15665 {
	static int n,m;
	static int[] arr,print;
	// 입력 순서를 지키기 위해 HashSet이 아닌 LinkedHashSet을 사용해야 한다. 
	static LinkedHashSet<String> set=new LinkedHashSet<>();
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		m=sc.nextInt();
		arr=new int[n];
		print=new int[m];
		
		for(int i=0;i<n;i++)
			arr[i]=sc.nextInt();
		
		Arrays.sort(arr); // 정렬된 배열 사용
		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;
		}
		
		// 숫자를 중복하여 선택해도 되기 때문에 visited 배열이나 for문의 시작을 지정하지 않아도 됨
		for(int i=0;i<n;i++) {
			print[depth]=arr[i];
			dfs(depth+1);
		}
	}
}

 

[고찰]

 N과 M 시리즈 11번째 문제이다. 이번 문제는 이전 10번째 문제에서 선택할 수 있는 수에 중복이 가능하나는 조건만 변경된 것으로 n개의 숫자중 m개의 숫자를 선택하는 부분만 변경하면 정답 처리를 받을 수 있는 간단한 문제였다. 이번 문제 또한 만들어진 수열이 오름차순이어야 하기 때문에 LinkedHashSet을 사용해야 한다는 점을 주의해야 한다.