Submission #92789

#	Time	Username	Problem	Language	Result	Execution time	Memory
92789		Vardanyan	Gift (IZhO18_nicegift)	C++14	49 / 100	1348 ms	120272 KiB

nicegift

#include<bits/stdc++.h>

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

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

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:47:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(long long i = 0;i<add.size();i++){
                             ~^~~~~~~~~~~
nicegift.cpp:55:31: warning: format '%lld' expects argument of type 'long long int', but argument 2 has type 'std::vector<std::vector<long long int> >::size_type {aka long unsigned int}' [-Wformat=]
     printf("%lld\n",ans.size());
                     ~~~~~~~~~~^
nicegift.cpp:56:26: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(long long i = 0;i<ans.size();i++){
                         ~^~~~~~~~~~~
nicegift.cpp:58:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(long long 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("%lld%lld",&n,&k);
     ~~~~~^~~~~~~~~~~~~~~~~~
nicegift.cpp:14:18: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
             scanf("%lld",&a[i].first);
             ~~~~~^~~~~~~~~~~~~~~~~~~~

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