Submission #523461

#	Time	Username	Problem	Language	Result	Execution time	Memory
523461		RaresFelix	Gift (IZhO18_nicegift)	C++17	49 / 100	2036 ms	117008 KiB

nicegift

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")

#define MN 1207171

using namespace std;
using ll = long long;

ll n, k, A[MN];
set<pair<ll, ll> > S;
vector<pair<ll, vector<ll>>> O;
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin >> n >> k;
    ll sum = 0;
    for(ll i = 1; i <= n; ++i)
        cin >> A[i], S.insert({A[i], i}), sum += A[i];
    if(S.rbegin()->first > sum / k || sum % k) {
        cout << "-1\n";
        return 0;
    }
    ll vmin, suma_ramasa, delta;
    vector<ll> PU;
    while(!S.empty()) {
        suma_ramasa = sum;
        for(ll i = 1; i <= k; ++i)
            PU.push_back(S.rbegin()->second), suma_ramasa -= S.rbegin()->first, S.erase(*S.rbegin());
        if(!S.empty())
            delta = min(A[PU.back()], sum / k - S.rbegin()->first);
        else delta = A[PU.back()];
        O.push_back({delta, PU});
        for(auto it : PU){
            A[it] -= delta;
            sum -= delta;
            if(A[it])
                S.insert({A[it], it});
        }
        PU.clear();
    }
    cout << O.size() << "\n";
    for(auto [c, v] : O) {
        cout << c << " ";
        for(auto it : v)
            cout << it << " ";
        cout << "\n";
    }
    return 0;
}

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:24:8: warning: unused variable 'vmin' [-Wunused-variable]
   24 |     ll vmin, suma_ramasa, delta;
      |        ^~~~

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