Submission #850637

#	Time	Username	Problem	Language	Result	Execution time	Memory
850637		vjudge1	A Difficult(y) Choice (BOI21_books)	C++17	45 / 100	1 ms	1364 KiB

books

#include <bits/stdc++.h>
#include "books.h"

using namespace std;

// typedef long long l;

vector<int> get_ans(vector<bool>& in){
    vector<int> ans;
    for(int i = 0; i < (int)in.size(); i++){
        if(in[i]) ans.push_back(i+1);
    }
    return ans;
}

void solve(int N, int K, long long A, int S) {
    vector<long long> v(N);
    vector<bool> in(N, 0);
    long long sum = 0;
    for(int i = 0; i < K; i++){
        v[i] = skim(i+1);
        in[i] = 1;
        sum += v[i];
    }
    if(sum > 2 * A) {impossible(); return;}
    if(sum >= A){
        answer(get_ans(in));
        return;
    }
    if(K == N) {impossible(); return;}
    int ind = N - 1;
    for(int i = K - 1; i >= 0; i--){
        v[ind] = skim(ind + 1);
        if(sum - v[i] + v[ind] > 2 * A){
            int lo = i + 1;
            int hi = ind - 1;
            sum -= v[i];
            in[i] = 0;
            while(lo <= hi){
                int mid = (lo + hi) / 2;
                v[mid] = skim(mid + 1);
                if(sum + v[mid] >= A && sum + v[mid] <= 2 * A){
                    in[mid] = 1;
                    answer(get_ans(in));
                    return;
                }else if(sum + v[mid] < A){
                    lo = mid + 1;
                }else{
                    hi = mid - 1;
                }
            }
            impossible();
            return;
        }else{
            sum -= v[i];
            sum += v[ind];
            in[i] = 0;
            in[ind] = 1;
            if(sum >= A){
                answer(get_ans(in));
                return;
            }
        }
        ind--;
    }
    impossible();
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 15
• Subtask 3: 10 / 10
• Subtask 4: 15 / 15
• Subtask 5: 0 / 15
• Subtask 6: 20 / 20
• Subtask 7: 0 / 20