[백준 / C++] 6603번 - "로또" 풀이

문제 설명

 

독일 로또는 {1, 2, ..., 49}에서 수 6개를 고른다.

 

로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다.

 

예를 들어, k=8, S={1,2,3,5,8,13,21,34}인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2,3,5,13,21], ..., [3,5,8,13,21,34])

 

집합 S와 k가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.

 

 

문제 주소: https://www.acmicpc.net/problem/6603

 

 

 

 

 

문제 풀이

 

k개의 원소를 갖는 집합 S에서 6개의 원소를 고르는 모든 경우의 수를 구하면 된다. 조합(n개의 원소 중 k개의 원소를 고르는 방법)을 이용하면 되는데 조합을 구하는 방법은 아래 링크로 들어가면 확인 할 수 있다. 

 

https://mjmjmj98.tistory.com/38

[C++ / Algorithm] 순열(next_permutation) 사용 방법과 조합(Combination) 구하기

 

 

소스코드

 

// 백준 6603: 로또
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int k;
vector<int> s(14);
 
int main() {
    while (true) {
        cin >> k;
        if (k == 0)    // 입력 종료
            break;
 
        // 배열 s 초기화, 조합 탐색에 사용할 v 초기화
        for (int i = 0; i < k; ++i)
            cin >> s[i];
        vector<int> v(6, 1);
        for (int i = 0; i < k - 6; ++i)
            v.push_back(0);
 
        // 모든 조합 탐색
        do {
            for (int i = 0; i < v.size(); ++i)
                if (v[i] == 1)
                    cout << s[i] << ' ';
            cout << endl;
        } while (prev_permutation(v.begin(), v.end()));
 
        cout << endl;
    }
 
    return 0;
}

 

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)