[BOJ 15655 N과 M(6)]

박현우·2020년 7월 6일
0

BOJ

목록 보기
6/87

N과M(6)

> 특이 사항

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

> 문제 풀이

N과M(2)와 N과M(5)를 적절하게 섞어서 풀면 된다.

  • Arrays.sort의 시간복잡도는 Dual Pivot Quick Sort를 사용하기 때문에 O(nlogn)을 따르거나 최악의 경우 O(n^2)을 따른다. 때에 따라 mergesort를 사용하는 식으로 시간 복잡도를 줄일 수 있다.

> 프로그램 코드(java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class BOJ15655_N과M6 {
	static int N, M;
	static boolean[] visited;
	static int[] a; // 실제 출력값을 담을 배열
	static int[] input;

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

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

		visited = new boolean[N];
		a = new int[N];
		input = new int[N];

		temp = br.readLine().split(" ");
		for (int i = 0; i < N; i++) {
			input[i] = Integer.parseInt(temp[i]);
		}
		Arrays.sort(input);
		dfs(0,input[0]);
	}

	static void dfs(int depth, int com) {
		if (depth == M) {// 종료 지점.
			for (int i = 0; i < M; i++) {
				System.out.print(a[i] + " ");
			}
			System.out.println();
			return;
		}

		for (int i = 0; i < input.length; i++) {
			if (!visited[i] && com <= input[i]) {
				visited[i] = true;
				a[depth] = input[i];
				dfs(depth + 1,input[i]);
				visited[i] = false;
			}
		}
	}
}

0개의 댓글