(C++) [백준 6603번] 로또
https://www.acmicpc.net/problem/6603
6603번: 로또
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로
www.acmicpc.net
1~49번까지의 번호가 있는 로또중에서 k개를 뽑아서(6<k<13) 숫자 k개로 구성된 집합 S를 만들고, 그 집합 S에 있는 숫자중에서 번호 6개를 뽑는 경우의 수를 모두 출력하는 문제이다.
예를 들어 k = 7, S = {1,2,3,4,5,6,7}이라면 번호를 뽑는 경우의 수는 총 7가지로 아래와 같다.
1 2 3 4 5 6
1 2 3 4 5 7
1 2 3 4 6 7
1 2 3 5 6 7
1 2 4 5 6 7
1 3 4 5 6 7
2 3 4 5 6 7재귀를 쓰는 문제인데 아직 익숙하지가 않아서 어렵게 느껴졌다.
우선 오름차순으로 출력해야 하는데, 집합 S에 원소가 들어올 때 오름차순으로 들어오므로 집합 S에 있는 순서대로 뽑는다고 가정하면 따로 정렬할 필요는 없다.
나는 선택한다 / 선택하지 않는다의 2가지 경우로 생각해서 모든 경우를 탐색할 수 있게 코드를 작성했다.
예를 들어, S = {1,2,3,4,5,6,7}이라고 가정하면 1을 선택한다 / 선택하지 않는다의 2가지 경우가 있고, 그 중에서 1을 선택하는 경우에서는 다시 2를 선택한다 / 선택하지 않는다... 그리고 마지막 7을 선택한다 / 선택하지 않는다에서 7을 선택하지 않는다까지 완료되면 코드가 종료되는 식이다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
#include <iostream>
#include <vector>
using namespace std;
int k;
vector <int> S;
void lotto(vector<int> v, int pos)
{
int i = 1;
if (k <= pos) return;
v.push_back(S[pos]);
if (v.size() == 6)
{
for (int i = 0; i < 6; i++)
cout << v[i] << ' ';
cout << "\n";
return;
}
while ((pos + i) < k)
lotto(v, pos + i++);
}
int main()
{
cin.tie(0); ios::sync_with_stdio(0);
cin >> k;
vector <int> v;
while (k != 0)
{
int t;
for (int i = 0; i < k; i++)
{
cin >> t;
S.push_back(t);
}
for (int i=0; i<k; i++)
lotto(v, i);
cout << "\n";
S.clear();
cin >> k;
}
}
|
cs |
벡터 S는 집합 S를 나타내고 pos는 집합 S 내에서 다음으로 선택할 번호의 위치이다. S의 pos 위치에 해당하는 원소를 하나 선택하고 6개가 되었을 때 출력하고 종료한다. 그렇지 않은 경우에는 반복문을 돌리는데 이 때 (pos + i)의 크기는 k보다 작아야한다. pos+i는 내가 다음으로 갈 위치 ( = 인덱스)에 해당하는데 pos+i가 k와 같거나 k보다 큰 경우 S의 범위를 초과하기 때문이다.
