알고리즘
백준_6603_로또
shj718
2022. 4. 12. 17:19
문제: https://www.acmicpc.net/problem/6603
6603번: 로또
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로
www.acmicpc.net
알고리즘: 브루트포스 - 재귀
추가로 알게된 사실: 함수의 매개변수로 vector를 넘겨줄 시 & 붙이는게 안전함.
코드:
#include <iostream>
#include <vector>
using namespace std;
vector<int> six;
void solve(vector<int> &s, int index, int cnt) { // index: 선택할지 말아야할지 결정해야하는 인덱스 / cnt: 현재까지 선택한 개수
// 정답을 찾은 경우
if (cnt == 6) {
for (int i = 0; i < 6; i++) {
cout << six[i] << ' ';
}
cout << '\n';
return;
}
// 불가능한 경우
if (index == s.size()) return;
// 다음 호출
six.push_back(s[index]);
solve(s, index + 1, cnt + 1);
six.pop_back();
solve(s, index + 1, cnt);
}
int main() {
while (1) {
int k;
cin >> k;
if (k == 0) break;
vector<int> s(k); // 사이즈 미리 안정하고 인덱스 접근시 에러 발생
for (int i = 0; i < k; i++) {
cin >> s[i];
}
solve(s, 0, 0);
cout << '\n';
}
return 0;
}