Submission #1018581

#	Time	Username	Problem	Language	Result	Execution time	Memory
1018581		overwatch9	A Difficult(y) Choice (BOI21_books)	C++17	45 / 100	2 ms	1192 KiB

books

#include <bits/stdc++.h>

#include "books.h"
// 
using namespace std;
using ll = long long;
//
// --- 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.
//


// vector <int> NUMS;
// ll skim(int x) {
//     return NUMS[x];
// }

void solve(int N, int K, ll A, int S) {
    vector <int> ans(K+1);
    ll sum = 0;
    vector <ll> nums(N+1);
    for (int i = 1; i <= K; i++) {
        ans[i] = i;
        nums[i] = skim(i);
        sum += nums[i];
    }
    for (int i = K; i >= 1 && sum < A; i--) {
        int lo = K+1, hi = N-(K-i);
        sum -= nums[i];
        if (nums[N - (K-i)] == 0)
            nums[N - (K-i)] = skim(N - (K-i));
        if (nums[N - (K-i)] + sum < A) {
            sum += nums[N - (K-i)];
            ans[i] = N - (K-i);
            continue;
        }
        while (lo < hi) {
            int mid = lo + (hi - lo) / 2;
            ll x = 0;
            if (nums[mid] != 0)
                x = nums[mid];
            else {
                nums[mid] = skim(mid);
                x = nums[mid];
            }
            if (sum + x >= A) {
                if (sum + x <= 2*A) {
                    lo = mid;
                    break;
                }
                hi = mid;
            } else
                lo = mid + 1;
        }
        if (nums[lo] == 0) {
            nums[lo] = skim(lo);
        }
        sum += nums[lo];
        ans[i] = lo;
    }
    if (sum > 2*A) {
        impossible();
    } else {
        vector <int> ans2(K);
        for (int i = 0; i < K; i++)
            ans2[i] = ans[i+1];
        answer(ans2);
    }

}
// int main() {
//     int n, k, a, s;
//     cin >> n >> k >> a >> s;
//     NUMS.resize(n+1);
//     for (int i = 1; i <= n; i++)
//         cin >> NUMS[i];
//     solve(n, k, a, s);
// }

• 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