로또 (같은 문자를 포함한 순열)

https://www.acmicpc.net/problem/6603

6603번: 로또

조합 = 같은 숫자가 포함된 순열

For Example

10개의 숫자에서 6개를 고른다?

그럼 0 6개와 1 4개를 같은 숫자가 포함된 순열로 만들면 됨!

마지막으로 0인 배열의 인덱스를 출력

 

각 테스트 케이스마다 수를 고르는 모든 방법을 출력한다.

이때, 사전 순으로 출력한다.

2차원 벡터에 저장하고 sort 한다.

 

<소스 코드>

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main() {
    while (true) 
    {
        int n;
        cin >> n;
        if (n == 0) 
            break;
       
        vector<int> a(n);
        for (int i = 0; i < n; i++) {
            cin >> a[i];
        }
        vector<int> d;
        for (int i = 0; i < n - 6; i++) {
            d.push_back(0);
        }
        for (int i = 0; i < 6; i++) {
            d.push_back(1);
        }
 
        vector<vector<int>> ans; // vector 2차원 배열
        do {
            vector<int> current;
            for (int i = 0; i < n; i++) {
                if (d[i] == 1) {
                    current.push_back(a[i]);
                }
            }
            ans.push_back(current);
        } while (next_permutation(d.begin(), d.end()));
 
        sort(ans.begin(), ans.end());
        for (auto& v : ans) {
            for (int i = 0; i < v.size(); i++) {
                cout << v[i] << ' ';
            }
            cout << '\n';
        }
        cout << '\n';
    }
    return 0;
}