Submission #334448

#	Time	Username	Problem	Language	Result	Execution time	Memory
334448		juggernaut	Gift (IZhO18_nicegift)	C++14	30 / 100	273 ms	8720 KiB

nicegift

#include<bits/stdc++.h>
using namespace std;
int n,k,mx,sum;
priority_queue<pair<int,int>>q;
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        q.push({x,i});
        mx=max(mx,x);
        sum+=x;
    }
    if(sum%k!=0||(mx<<1)>sum)return puts("-1"),0;
    vector<vector<int>>ans;
    while(!q.empty()){
        vector<int>v;
        stack<pair<int,int>>temp;
        for(int i=0;i<k;i++){
            temp.push(q.top());
            q.pop();
            v.push_back(temp.top().second);
            if(temp.top().first<2)temp.pop();
            else temp.top().first--;
        }
        while(!temp.empty()){
            q.push(temp.top());
            temp.pop();
        }
        ans.push_back(v);
    }
    printf("%d",(int)ans.size());
    for(auto v:ans){
        printf("\n1");
        for(auto to:v)printf(" %d",to);
    }
}

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:6:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
    6 |     scanf("%d%d",&n,&k);
      |     ~~~~~^~~~~~~~~~~~~~
nicegift.cpp:9:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
    9 |         scanf("%d",&x);
      |         ~~~~~^~~~~~~~~

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