[BOJ 15666 N과M(12)]

박현우·2020년 7월 10일
0

BOJ

목록 보기
12/87

N과M(12)

> 특이 사항

  • N개의 자연수 중에서 M개를 고른 수열.
  • 같은 수를 여러 번 골라도 된다.
  • 고른 수열은 비내림차순이어야 한다.
    길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

> 문제 풀이

  • N과M 시리즈의 마지막 문제이다. 특별한 것은 없고 10번과 11번에서 사용한 것들을 적절히 사용한다.

> 프로그램 코드(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.Set;
import java.util.TreeSet;

public class BOJ15666_N과M12 {
	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, "", a.get(0));
		bw.flush();
		bw.close();
		br.close();
	}

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

		for (int i = 0; i < a.size(); i++) {
			if (com <= a.get(i)) {
				dfs(depth + 1, bw, str + a.get(i) + " ", a.get(i));
			}
		}
	}
}

0개의 댓글