[백준/C++] 6603번: 로또

문제

풀이

k개의 수가 주어지고, 이 수들 중에서 6개의 무작위 조합을 뽑아내는 문제이다.

6개의 1과 k - 6개의 0을 가진 배열 sub을 만들고 이 sub 배열로 순열을 만든다.

sub 배열에서 값이 1인 인덱스를 가져와 입력 받은 로또 번호를 출력해주어 순열로 조합을 구현하였다.

구현

#include <bits/stdc++.h>
using namespace std;

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

	while (1) {
		int K; cin >> K;
		if (K == 0) break;

		vector<int> lotto(K);

		for (int i = 0; i < K; i++) {
			cin >> lotto[i];
		}

		vector<int> sub;

		for (int i = 0; i < 6; i++) sub.push_back(0);
		for (int i = 0; i < K - 6; i++) sub.push_back(1);


		do {
			for (int i = 0; i < K; i++) {
				if (sub[i] == 0) {
					cout << lotto[i] << ' ';
				}
			}
			cout << '\n';

		} while (next_permutation(sub.begin(), sub.end()));

		cout << '\n';
	}


	
	return 0;
}