# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1100551 | epicci23 | A Difficult(y) Choice (BOI21_books) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "bits/stdc++.h"
#include "books.h"
using namespace std;
//#define int long long
#define all(v) v.begin(),v.end()
#define sz(v) (int)v.size()
void solve(int N, int K, long long A, int S) {
vector<long long> ilk;
for(int i=1;i<=k;i++) ilk.push_back(skim(i));
long long summ=accumulate(all(ilk),0LL);
if(summ>2*A){
impossible();
return;
}
if(summ>=A){
vector<int> lol;
for(int i=1;i<=k;i++) lol.push_back(i);
answer(lol);
return;
}
int l=k,r=n;
while(l<r){
int mid=(l+r+1)/2;
if(skim(mid)<A) l=mid;
else r=mid-1;
}
int kaydet1=skim(l);
int kaydet2=-1;
if(l+1<=n) kaydet2=skim(l+1);
summ-=ilk[k-1];
summ+=kaydet1;
if(summ>=A && summ<=2*A){
vector<int> lol;
for(int i=1;i<k;i++) lol.push_back(i);
lol.push_back(l);
answer(lol);
return;
}
if(kaydet2==-1){
impossible();
return;
}
summ-=kaydet1;
summ+=kaydet2;
if(summ>=A && summ<=2*A){
vector<int> lol;
for(int i=1;i<k;i++) lol.push_back(i);
lol.push_back(l+1);
answer(lol);
return;
}
impossible();
return;
}
/*
void _(){
}
int32_t main(){
cin.tie(0);
ios::sync_with_stdio(0);
int tc=1;//cin >> tc;
while(tc--) _();
}*/