Submission #90946

#	Time	Username	Problem	Language	Result	Execution time	Memory
90946		toloraia	Gift (IZhO18_nicegift)	C++14	49 / 100	994 ms	220960 KiB

nicegift

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define F first
#define S second

using namespace std;

const ll N = 2e6 + 5, INF = 2e18 + 5;

ll n, k;
pair < ll, ll > P[N];
ll nn;
vector < ll > V[N], v;

main()
{
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for (ll i = 1; i <= n; i++){
        cin>>P[i].F;
        P[i].S = i;
    }
    sort (P + 1, P + n + 1);
    reverse (P + 1, P + n + 1);
    if (P[1].F == P[n].F){
        ll d = __gcd (n, k);
        ll m = k / d;
        if (P[1].F % m != 0){
            cout<<-1<<endl;
            return 0;
        }
        ll p = P[1].F / m;
        cout<<n / d<<endl;
        for (ll i = 1; i <= n; i += d){
            cout<<p<<" ";
            for (ll j = i; j < i + k; j++)
                cout<<(j - 1) % n + 1<<" ";
            cout<<endl;
        }
        return 0;
    }
        while (P[k].F > 0){
            ll I1 = k, I2 = k;
            for (ll i = 20; i >= 0; i--){
                if (I1 - (1<<i) <= 0)
                    continue;
                if (P[I1 - (1<<i)].F == P[k].F)
                    I1 -= (1<<i);
            }
            for (ll i = 20; i >= 0; i--){
                if (I2 + (1<<i) > n)
                    continue;
                if (P[I2 + (1<<i)].F == P[k].F)
                    I2 += (1<<i);
            }
            ll x = min (max (1ll, P[I1 - 1].F - P[I1].F), P[I2].F - P[I2 + 1].F);
            v.clear();
            v.pb(x);
            for (ll i = 1; i < I1; i++){
                v.pb (P[i].S);
                P[i].F-=x;
            }
            for (ll i = I2; (ll)v.size() <= k; i--){
                v.pb (P[i].S);
                P[i].F-=x;
            }
            V[++nn] = v;
        }
        if (P[1].F > 0){
            cout<<-1<<endl;
            return 0;
        }
        cout<<nn<<endl;
        for (ll i = 1; i <= nn; i++){
            cout<<V[i][0]<<" ";
            for (ll j = 1; j <= k; j++)
                cout<<V[i][j]<<" ";
            cout<<endl;
        }
    return 0;
}

Compilation message (stderr)

nicegift.cpp:16:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main()
      ^

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