알고리즘/BOJ

[BOJ] 15654. N과 M (5)

hyerann 2019. 4. 17. 13:32

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

 

15654번: N과 M (5)

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

www.acmicpc.net

N과 M (5)는 앞의 문제들과 다르게 N까지의 자연수가 아닌 N개의 자연수를 입력받고 그 수들로 수열을 만드는 것이다.

입력받은 수를 numbers[] 배열에 넣었고, visited[] 배열을 사용하여 해당 수가 쓰인지 쓰이지않은지 체크하였다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다는 조건이 있으므로, <algorithm> 헤더 파일의 sort() 함수를 사용하여 입력받은 수를 오름차순으로 정렬한 후 수열을 만들어주었다.

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

int N, M, numbers[8], visited[8];
vector<int> series;
void dfs(int cnt);
void print();

int main() {
	cin.tie(0); ios::sync_with_stdio(0);
	cin >> N >> M;
	for (int i = 0; i < N; i++) {
		cin >> numbers[i];
	}
	sort(numbers, numbers + N);
	dfs(0);
	return 0;
}

void dfs(int cnt) {
	if (cnt == M) {
		print();
		return;
	}

	for (int i = 0; i < N; i++) {
		if (!visited[i]) {
			visited[i] = true;
			series.push_back(numbers[i]);
			dfs(cnt + 1);
			visited[i] = false;
			series.pop_back();
		}
	}
}

void print() {
	for (int i = 0; i < M; i++) {
		cout << series[i] << ' ';
	}
	cout << "\n";
}