Submission #378735

#	Time	Username	Problem	Language	Result	Execution time	Memory
378735		i_am_noob	Gift (IZhO18_nicegift)	C++17	49 / 100	440 ms	58144 KiB

nicegift

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

#define int long long
#define rep(n) rep1(i,n)
#define rep1(i,n) rep2(i,0,n)
#define rep2(i,a,b) for(int i=a; i<b; ++i)
#define rep3(i,a,b) for(int i=a; i>=b; --i)
#define sz(a) ((int)a.size())
#define all(a) a.begin(),a.end()
#define pow2(x) (1ll<<(x))
#define pii pair<int,int>
#define pb push_back

#ifdef i_am_noob
#define bug(...) cerr << "#" << __LINE__ << ' ' << #__VA_ARGS__ << "- ", _do(__VA_ARGS__)
template<typename T> void _do(T && x){cerr << x << endl;}
template<typename T, typename ...S> void _do(T && x, S&&... y){cerr << x << ' ',_do(y...);}
#else
#define bug(...) 826
#endif

const int maxn=1000005;
const int mod=1000000007;

int n,k,a[maxn],cur;
priority_queue<pii> pq;
vector<vector<int>> res;

signed main(){
    ios_base::sync_with_stdio(0),cin.tie(0);
    #ifdef i_am_noob
    freopen("input1.txt","r",stdin);
    freopen("output1.txt","w",stdout);
    freopen("output2.txt","w",stderr);
    #endif
    cin >> n >> k;
    rep(n) cin >> a[i];
    int sum=0;
    rep(n) sum+=a[i];
    if(sum<=100000){
        rep(n) pq.push({a[i],i});
        while(sz(pq)>=k){
            vector<int> vec;
            rep(k){
                auto [x,y]=pq.top();
                pq.pop();
                vec.pb(y);
            }
            res.pb({1});
            for(auto i: vec) res.back().pb(i+1);
            for(auto i: vec) a[i]--;
            for(auto i: vec) if(a[i]) pq.push({a[i],i});
        }
        if(!pq.empty()){
            cout << "-1\n";
            return 0;
        }
        cout << sz(res) << "\n";
        for(auto& vec1: res){
            for(auto i: vec1) cout << i << ' ';
            cout << "\n";
        }
        return 0;
    }
    cur=0;
    if(a[0]*n%k){
        cout << "-1\n";
        return 0;
    }
    rep(n){
        res.pb({0});
        rep1(j,k) res.back().pb(cur+1),cur=(cur+1)%n;
        if(cur==0) break;
    }
    for(auto& vec1: res) vec1[0]=a[0]*n/k/sz(res);
    cout << sz(res) << "\n";
    for(auto& vec1: res){
        for(auto i: vec1) cout << i << ' ';
        cout << "\n";
    }
}

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