제출 #414977

#제출 시각아이디문제언어결과실행 시간메모리
414977tengiz05A Difficult(y) Choice (BOI21_books)C++17
0 / 100
3 ms968 KiB
#include <bits/stdc++.h>
#include "books.h"
using i64 = long long;
void solve(int n, int k, i64 A, int s) {
    std::vector<i64> a(n, -1);
    auto ask = [&](int i) -> i64 {
        if (~a[i]) {
            return a[i];
        } else {
            return a[i] = skim(i + 1);
        }
    };
    int l = -1, r = n;
    while (l + 1 < r) {
        int mid = (l + r) / 2;
        if (ask(mid) * k > A && ask(mid) * k <= 2 * A) {
            l = mid;
            break;
        } else if (ask(mid) * k <= A) {
            l = mid;
        } else {
            r = mid;
        }
    }
    int t = std::max(0, l - k);
    std::vector<int> ans;
    i64 sum = 0;
    for (int i = 0; i < k - 1; i++) {
        sum += ask(i + t);
        ans.push_back(i + t + 1);
    }
    // std::cout << sum << "\n";
    l = t + k - 2, r = n;
    while (l + 1 < r) {
        int mid = (l + r) / 2;
        if (sum + ask(mid) >= A && sum + ask(mid) <= 2 * A) {
            ans.push_back(mid);
            answer(ans);
        } else if (sum + ask(mid) < A) {
            l = mid;
        } else {
            r = mid;
        }
    }
    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...