[BOJ] 15650. N과 M(2) (C++)

• 코딩 초보

---

15650. N과 M(2) (실버3)

 

문제

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

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

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

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

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

int n, m;
vector <int> v;
int chk[9] = { 0, };
//큰 수를 우선적으로 선택해야 


void func(int k,int tmp) {
	if (k == m) {
		for (int i = 0; i < m; i++) {
			cout << v[i] << " ";
		}
		cout << "\n";
		return; 
	}
	for (int i = 1; i <= n; i++) {
		if (tmp > i) continue; 
		if (chk[i] == 1) continue; 
		chk[i] = 1; 
		v.push_back(i);

		func(k + 1, i); 

		v.pop_back();
		chk[i] = 0; 
	}

}

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

	cin >> n >> m;

	func(0,1);


	return 0;
}

저번 N과 M(1)에 제출했던 코드를 변형시켰다. 오름차순으로 출력하기 위해선 항상 큰 수를 선택하도록 해야한다. 

그래서 매개변수 tmp를 추가하여 현재 수를 알 수 있도록 하고, 현재 수보다 큰 수를 선택하도록 하였다. 만약 현재수(tmp)보다 작은 수의 경우는 continue를 통해 아래 코드들을 실행하지 않도록 해주면 된다. 그 외의 부분은 N과 M(1)에서 풀었던 코드와 비슷하게 진행된다.