Submission #171142

#	Time	Username	Problem	Language	Result	Execution time	Memory
171142		juggernaut	Gift (IZhO18_nicegift)	C++14	30 / 100	2041 ms	196728 KiB

nicegift

//Just try and the idea will come!
#include<bits/stdc++.h>
#define int long long int
using namespace std;
priority_queue<pair<int,int>>q;
int n,k,i,x;
pair<int,int>a[100000];
vector<vector<int>>ans;
main(){
    scanf("%lld%lld",&n,&k);
    for(i=0;i<n;i++){
        scanf("%lld",&x);
        q.push({x,i+1});
    }
    while(!q.empty()){
        if(q.size()<k){
            puts("-1");
            return 0;
        }
        vector<int>res;
        for(i=0;i<k;i++)a[i]=q.top(),q.pop();
        for(i=0;i<k;i++){
            res.push_back(a[i].second);
            if(a[i].first!=1)q.push({a[i].first-1,a[i].second});
        }
        ans.push_back(res);
    }
    printf("%lld\n",(int)ans.size());
    for(auto res:ans){
        printf("1 ");
        for(int to:res)printf("%lld ",to);
        puts("");
    }
}

Compilation message (stderr)

nicegift.cpp:9:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^
nicegift.cpp: In function 'int main()':
nicegift.cpp:16:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if(q.size()<k){
            ~~~~~~~~^~
nicegift.cpp:10:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld%lld",&n,&k);
     ~~~~~^~~~~~~~~~~~~~~~~~
nicegift.cpp:12:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld",&x);
         ~~~~~^~~~~~~~~~~

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