Submission #1365894

#	Time	Username	Problem	Language	Result	Execution time	Memory
1365894		lucasdmy	A Difficult(y) Choice (BOI21_books)	C++20	In queue	0 ms	0 KiB

books

#include <bits/stdc++.h>
using namespace std;
void solve(int n, int books, long long int a, int s){
    vector<long long int>v(n+10);
    long long int sum=0;
    vector<int>resp;
    for(int k=1;k<books;k++){
        v[k]=skim(k);
        sum+=v[k];
        res.push_back(k);
    }
    int p1=books, p2=n;
    while(p1<p2){
        int m=(p1+p2+1)/2;
        v[m]=skim(m);
        if(a<=v[m]+sum and v[m]+sum<=2*a){
            resp.push_back(m);
            answer(resp);
        }
        if(v[m]+sum<a){
            p1=m;
        }else{
            p2=m-1;
        }
    }
    resp.push_back(p1);
    sum+=v[p1];
    for(int k=p1-1;k>=p1-books+1;k--){
        int aux=skim(k);
        int idx=books-(p1-k)-1;
        resp[idx]=k;
        sum+=aux-v[idx];
        if(sum>=a){
            answer(resp);
        }
    }
    impossible();
}