[문과 코린이의 IT 기록장] C++ 백준 문제풀이[BF] - N과 M(5) (15654)

[문과 코린이의 IT 기록장] C++ 백준 문제풀이[BF] - N과 M(5) (15654)

[문과 코린이의 IT 기록장] C++ 백준 문제풀이[BF] - N과 M(5) (15654)

 

---

 

15654번: N과 M (5)

[ 문제 ]

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

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

[ 입력 ]

• 첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
• 둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

[ 출력 ]

• 한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
• 수열은 사전 순으로 증가하는 순서로 출력해야 한다.

---

[ 코드 ]

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

int a[10];
bool chk[10];
int seq[10];
//vector <int> seq(10);

void go(int index, int n, int m) {
	if (index == m)
	{
		for (int i = 0; i < m; i++)
		{
			cout << seq[a[i]]; // seq값을 a배열을 통해 출력
			if (i != m - 1)
			{
				cout << ' ';
			}
		}
		cout << '\n';
		return;
	}

	for (int i = 0; i < n; i++)
	{
		if (chk[i])
		{
			continue;
		}
		a[index] = i; // seq배열의 인덱스 값을 a배열에 저장
		chk[i] = true;
		go(index + 1, n, m);
		chk[i] = false;
	}

}

int main() {
	int N, M;
	cin >> N >> M;

	for (int i = 0; i < N; i++)
	{
		cin >> seq[i];
	}
	sort(seq, seq + N); // vector말고 배열로 선언되야 출력됨
	// sort(seq.begin(), seq.end()) : 입력한 원소 외에 나머지 0으로 초기화된 것들도 함께 정렬한다.
	go(0, N, M);
	return 0;
}