제출 #168467

#제출 시각아이디문제언어결과실행 시간메모리
168467achibasadzishviliGift (IZhO18_nicegift)C++14
30 / 100
69 ms19960 KiB
#include <bits/stdc++.h> #define ll int #define f first #define s second #define pb push_back #define mp make_pair using namespace std; ll n,k,a[200005],s,las; set<pair<ll,ll> >st; vector<vector<ll> >ans; vector<ll>g; int main(){ ios::sync_with_stdio(false); cin >> n >> k; bool ok = 1; for(int i=1; i<=n; i++){ cin >> a[i]; if(a[i] != a[1])ok = 0; s += a[i]; st.insert(mp(-a[i] , i)); } /* if(ok){ ll m = n , ind = 1; while(m - k >= k){ m -= k; g.clear(); g.pb(a[ind]); for(int i=ind; i<ind + k; i++){ g.pb(i); a[i] = 0; } ind += k; ans.pb(g); } st.clear(); s = 0; for(int i=1; i<=n; i++){ s += a[i]; if(a[i]) st.insert(mp(-a[i] , i)); } } */ if(s % k){ cout << "-1"; return 0; } ll raod = s / k; ll sum = s; while(sum){ if((int)st.size() < k){ cout << "-1"; return 0; } ll p = 0,cur = 0; g.clear(); ll maxi = -(*st.begin()).f; for(set<pair<ll,ll> >::iterator it = st.begin(); it != st.end(); it++){ p++; if(p > k){ cur = (*it).s; break; } las = (*it).s; } ll l = 1 , r = a[las] , mid , can = 1; while(r >= l){ mid = (l + r) / 2; if(max(maxi - mid , a[cur]) * k <= sum - mid * k){ l = mid + 1; can = mid; } else { r = mid - 1; } } sum -= can * k; p = 0; g.pb(can); for(set<pair<ll,ll> >::iterator it = st.begin(); it != st.end(); it++){ p++; if(p > k){ cur = (*it).s; break; } g.pb((*it).s); } for(int i=1; i<g.size(); i++){ st.erase(mp(-a[g[i]] , g[i])); a[g[i]]-=can; if(a[g[i]])st.insert(mp(-a[g[i]] , g[i])); } ans.pb(g); } cout << (int)ans.size() << endl; for(int i=0; i<ans.size(); i++){ for(int j=0; j<ans[i].size(); j++) cout << ans[i][j] << " "; cout << '\n'; } return 0; }

컴파일 시 표준 에러 (stderr) 메시지

nicegift.cpp: In function 'int main()':
nicegift.cpp:91:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i=1; i<g.size(); i++){
                      ~^~~~~~~~~
nicegift.cpp:101:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0; i<ans.size(); i++){
                  ~^~~~~~~~~~~
nicegift.cpp:102:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int j=0; j<ans[i].size(); j++)
                      ~^~~~~~~~~~~~~~
nicegift.cpp:15:10: warning: variable 'ok' set but not used [-Wunused-but-set-variable]
     bool ok = 1;
          ^~
nicegift.cpp:51:8: warning: unused variable 'raod' [-Wunused-variable]
     ll raod = s / k;
        ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...