Submission #92768

#	Time	Username	Problem	Language	Result	Execution time	Memory
92768		Vardanyan	Gift (IZhO18_nicegift)	C++14	30 / 100	513 ms	55092 KiB

nicegift

#include<bits/stdc++.h>

using namespace std;
const int N = 1000*1000+5;
pair<int,int> a[N];

int main()   {
    int n,k;
    scanf("%d%d",&n,&k);
    int sum = 0;
    set<pair<int,int >,greater<pair<int,int> > > ms;
    for(int i = 1;i<=n;i++){
            scanf("%d",&a[i].first);
            a[i].second = i;
            sum+=a[i].first;
            ms.insert(a[i]);
    }
    sort(a+1,a+1+n);
    vector<vector<int> > ans;
    while(1){
        int kk = k;
        vector<int> now;
        set<pair<int,int> >::iterator it = ms.begin();
        vector<pair<int,int> > add;
        for(;it!=ms.end();it++){
            pair<int,int> x = *it;
            if(x.first<=0){
                printf("-1\n");
                return 0;
            }
            //x.first--;
            sum--;
            now.push_back(x.second);
            kk--;
            add.push_back(x);
            if(kk == 0) break;
        }
        for(int i = 0;i<add.size();i++){
                ms.erase(ms.find(add[i]));
                add[i].first--;
                ms.insert(add[i]);
        }
        ans.push_back(now);
        if(sum == 0) break;
    }
    printf("%d\n",ans.size());
    for(int i = 0;i<ans.size();i++){
        printf("1 ");
        for(int j = 0;j<ans[i].size();j++){
            printf("%d ",ans[i][j]);
        }
        printf("\n");
    }
    return 0;
}

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:38:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i = 0;i<add.size();i++){
                       ~^~~~~~~~~~~
nicegift.cpp:46:29: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::vector<std::vector<int> >::size_type {aka long unsigned int}' [-Wformat=]
     printf("%d\n",ans.size());
                   ~~~~~~~~~~^
nicegift.cpp:47:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0;i<ans.size();i++){
                   ~^~~~~~~~~~~
nicegift.cpp:49:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int j = 0;j<ans[i].size();j++){
                       ~^~~~~~~~~~~~~~
nicegift.cpp:9:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d",&n,&k);
     ~~~~~^~~~~~~~~~~~~~
nicegift.cpp:13:18: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
             scanf("%d",&a[i].first);
             ~~~~~^~~~~~~~~~~~~~~~~~

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