Submission #1093839

#TimeUsernameProblemLanguageResultExecution timeMemory
1093839TymondA Difficult(y) Choice (BOI21_books)C++17
60 / 100
2 ms868 KiB
#include <bits/stdc++.h>
#include "books.h"
using namespace std;
using ll = long long;
using ld = long double;
#define fi first
#define se second
#define vi vector<int>
#define vll vector<long long>
#define pii pair<int, int>
#define pll pair<long long, long long>
#define pb push_back
#define mp make_pair
#define eb emplace_back
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
mt19937_64 rng64(chrono::steady_clock::now().time_since_epoch().count());
inline int rand(int l,int r){return uniform_int_distribution<int>(l, r)(rng);}
inline ll rand(ll l,ll r){return uniform_int_distribution<ll>(l, r)(rng64);}
#ifdef DEBUG
auto&operator<<(auto&o,pair<auto,auto>p){return o<<"("<<p.first<<", "<<p.second<<")";}
auto operator<<(auto&o,auto x)->decltype(x.end(),o){o<<"{";int i=0;for(auto e:x)o<<","+!i++<<e;return o<<"}";}
#define debug(X...)cerr<<"["#X"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(X)
#else
#define debug(...){}
#endif
 
struct custom_hash {
    static uint64_t splitmix64(uint64_t x) {
        x += 0x9e3779b97f4a7c15;
        x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
        x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
        return x ^ (x >> 31);
    }
 
    size_t operator()(uint64_t x) const {
        static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
        return splitmix64(x + FIXED_RANDOM);
    }
};

//
// --- Sample implementation for the task books ---
//
// To compile this program with the sample grader, place:
//     books.h books_sample.cpp sample_grader.cpp
// in a single folder and run:
//     g++ books_sample.cpp sample_grader.cpp
// in this folder.
//

const int MAXN = 1e5 + 7;
int n, k;
ll a[MAXN];
ll A;

ll Get(int ind){
    if(a[ind] != 0){
        return a[ind];
    }
    a[ind] = skim(ind);
    return a[ind];
}

void solve(int N, int K, long long nA, int S) {
    n = N;
    k = K;
    A = nA;

    vi used(n + 1, 0);
    ll sum = 0LL;
    for(int i = 1; i <= k; i++){
        used[i] = 1;
        sum += Get(i);
    }

    if(sum > 2 * A){
        impossible();
        return;
    }

    if(sum >= A){
        vi ans = {};
        for(int i = 1; i <= k; i++){
            ans.pb(i);
        }
        answer(ans);
        return;
    }

    int f = n;
    for(int i = k; i >= 1; i--){
        int l = i;
        int p = f;
        int s;
        while(l < p){
            s = (l + p + 1) / 2;
            ll x = Get(s);
            if(sum - a[i] + x <= 2 * A){
                l = s;
            }else{
                p = s - 1;
            }
        }

        used[i] = 0;
        sum -= a[i];
        used[l] = 1;
        sum += a[l];

        if(sum >= A){
            vi ans = {};
            for(int i = 1; i <= n; i++){
                if(used[i]){
                    ans.pb(i);
                }
            }
            answer(ans);
            return;
        }

        f = l - 1;
    }
    impossible();
}
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...