Submission #1239487

#	Time	Username	Problem	Language	Result	Execution time	Memory
1239487		_callmelucian	Gift (IZhO18_nicegift)	C++17	7 / 100	223 ms	24628 KiB

nicegift

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ld = long double;
using pl = pair<ll,ll>;
using pii = pair<int,int>;
using tpl = tuple<int,int,int>;

#define all(a) a.begin(), a.end()
#define filter(a) a.erase(unique(all(a)), a.end())

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int n, k; cin >> n >> k;
    vector<ll> a(n);
    for (ll &u : a) cin >> u;

    ll sum = accumulate(all(a), 0LL);
    if (sum % k) return cout << -1 << "\n", 0;

    priority_queue<pii> pq;
    for (int i = 0; i < n; i++) pq.emplace(a[i], i);

    cout << sum / k << "\n";
    for (int i = 0; i < sum / k; i++) {
        cout << 1 << " ";
        vector<pii> wait;
        for (int j = 0; j < k; j++) {
            if (pq.empty()) return cout << -1 << "\n", 0;
            int a, idx; tie(a, idx) = pq.top(); pq.pop();
            cout << idx + 1 << " ";
            if (a > 1) wait.emplace_back(a - 1, idx);
        }
        for (auto [a, b] : wait) pq.emplace(a, b);
        cout << "\n";
    }

    return 0;
}

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