[BOJ 15663 N과M(9)]

N과M(9)

> 특이 사항

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

> 문제 풀이

• N과M(5)번과 다른점은 숫자가 주어지지만 중복된 숫자가 주어질 수 있다는 점이다. 중복된 숫자를 어떻게 처리하느냐가 관건이다.
• 중복을 허용하지않고 오름차순 정렬상태를 유지해야 한다.
• Set에서 파생된 TreeSet이 이와 같은 성질을 띄기 때문에 TreeSet을 이용한다.
• Set의 내용을 출력하려면 데이터를 담을 자료구조가 필요하다. Arraylist를 이용한다.

> 프로그램 코드(java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Set;
import java.util.TreeSet;

public class BOJ15663_N과M9 {
	static int N, M;
	static int[] visited = new int[10001];
	static ArrayList<Integer> a;
	static int[] input;

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		String[] temp = br.readLine().split(" ");// temp에 " "을 기준으로 String을 나눠 담음.
		N = Integer.parseInt(temp[0]); // String을 int로 변환 후 담음.
		M = Integer.parseInt(temp[1]);

		input = new int[N];

		temp = br.readLine().split(" ");
		Set<Integer> s = new TreeSet<Integer>();
		for (int i = 0; i < N; i++) {
			int t = Integer.parseInt(temp[i]);
			s.add(t); // TreeSet에 add
			visited[t]++; // 각 번호에 맞게 카운트
		}
		a = new ArrayList<Integer>(s); // arraylist에 트리셋을 넣어줌.
		dfs(0, bw, "");
		bw.flush();
		bw.close();
		br.close();
	}

	static void dfs(int depth, BufferedWriter bw, String str) {
		if (depth == M) {// 종료 지점.
			try {
				bw.write(str + "\n");
			} catch (IOException e) {
				e.printStackTrace();
			}
			return;
		}

		for (int i = 0; i < a.size(); i++) {
			if (visited[a.get(i)] > 0) {
				visited[a.get(i)]--;
				dfs(depth + 1, bw, str + a.get(i) + " ");
				visited[a.get(i)]++;

			}
		}
	}
}