Submission #608805

#	Time	Username	Problem	Language	Result	Execution time	Memory
608805		pakhomovee	Gift (IZhO18_nicegift)	C++17	30 / 100	349 ms	78592 KiB

nicegift

#include <iostream>
#include <vector>
#include <string>
#include <iomanip>
#include <algorithm>
#include <set>
#include <numeric>
#include <queue>

using namespace std;

int32_t main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (int& i : a) cin >> i;
    vector<int> b(n, 0);
    priority_queue<pair<int, int>> q;
    vector<pair<int, vector<int>>> act;
    for (int i = 0; i < n; ++i) q.push({ a[i], i });
    while (q.size() >= k) {
        vector<int> pos;
        for (int j = 0; j < k; ++j) {
            pos.push_back(q.top().second);
            q.pop();
        }
        for (int i : pos) {
            ++b[i];
            if (b[i] < a[i]) {
                q.push({ a[i] - b[i], i });
            }
        }
        act.push_back({ 1, pos });
    }
    if (q.empty()) {
        cout << act.size() << '\n';
        for (pair<int, vector<int>> x : act) {
            cout << x.first << ' ';
            for (int j : x.second) {
                cout << j + 1 << ' ';
            }
            cout << '\n';
        }
    } else {
        cout << -1;
    }
}

Compilation message (stderr)

nicegift.cpp: In function 'int32_t main()':
nicegift.cpp:23:21: warning: comparison of integer expressions of different signedness: 'std::priority_queue<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   23 |     while (q.size() >= k) {
      |            ~~~~~~~~~^~~~

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 12 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 51