Submission #950620

#	Time	Username	Problem	Language	Result	Execution time	Memory
950620		Pring	A Difficult(y) Choice (BOI21_books)	C++17	0 / 100	1 ms	344 KiB

books

#include <bits/stdc++.h>
#include "books.h"
using namespace std;

#define fs first
#define sc second
#define mp make_pair
#define FOR(i, j, k) for (int i = j, Z = k; i < Z; i++)
typedef pair<int, int> pii;
using ll = long long;

namespace {
    const int MXN = 10;
    int n, k, s;
    ll a;
    int b;
    vector<int> v;
    ll l[MXN], r[MXN];

    int BSH(int l, int r, int b) {
        while (l + 1 < r) {
            int mid = (l + r) >> 1;
            (skim(mid) >= b ? r : l) = mid;
        }
        return r;
    }

    void SOLVE() {
        b = BSH(0, n + 1, a);
        if (b < k) {
            impossible();
            return;
        }
        FOR(i, 0, k) l[i] = skim(i + 1);
        if (b != k) FOR(i, 0, k) r[i] = skim(b - k + i);
        if (b != n + 1) {
            ll c = skim(b);
            ll acc = 0;
            FOR(i, 0, k - 1) acc += l[i];
            acc += c;
            if (acc <= 2 * a) {
                FOR(i, 1, k) v.push_back(i);
                v.push_back(b);
                answer(v);
                return;
            }
        }
        if (b == k) {
            impossible();
            return;
        }
        ll acc = 0;
        FOR(i, 0, k) acc += r[i];
        if (acc < a) {
            impossible();
            return;
        }
        if (acc <= 2 * a) {
            FOR(i, 0, k) v.push_back(b - k + i);
            answer(v);
            return;
        }
        FOR(i, 0, k) {
            acc -= r[i];
            acc += l[i];
            if (acc <= 2 * a) {
                FOR(j, 0, i + 1) v.push_back(j);
                FOR(j, i + 1, k) v.push_back(b - k + i);
                answer(v);
                return;
            }
        }
    }
}

void solve(int n, int k, ll a, int s) {
    ::n = n;
    ::k = k;
    ::a = a;
    ::s = s;
    SOLVE();
}

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