[백준/BOJ] 백준 15654번 : N과 M (5)

www.acmicpc.net/problem/15654

15654번: N과 M (5)

입력받은 수를 오름차순으로 정렬하고, 그 순서대로 중복되지 않게 m개의 숫자를 고른다. 숫자를 고를 때는 고른 숫자는 체크를 하여 중복해서 숫자를 고르지 않도록 하였다.

 

코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int n, m;
vector<int> number;

void Solve(vector<int>& selected, vector<int>& check)
{
	//m개를 골랐을때
	if (selected.size() == m)
	{
		for (int i = 0; i < selected.size(); i++)
			cout << selected[i] << " ";
		cout << "\n";

		return;
	}

	for (int i = 0; i < n; i++)
	{
		//이미 고른 숫자일때
		if (check[i] == 1)
			continue;

		//고른 숫자를 추가, 고른숫자 체크
		selected.push_back(number[i]);
		check[i] = 1;

		Solve(selected, check);

		//고른 숫자 제거, 제거한 숫자 체크 해제
		selected.pop_back();
		check[i] = 0;
	}
}

int main()
{
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);

	int input;

	cin >> n >> m;

	for (int i = 0; i < n; i++)
	{
		cin >> input;
		number.push_back(input);
	}

	//입력받은 수를 오름차순으로 정렬한다
	sort(number.begin(), number.end());

	vector<int> selected;
	vector<int> check(n, 0);

	Solve(selected, check);

	return 0;
}