Submission #594933

#	Time	Username	Problem	Language	Result	Execution time	Memory
594933		Do_you_copy	A Difficult(y) Choice (BOI21_books)	C++17	50 / 100	1 ms	256 KiB

books

#include <bits/stdc++.h>
#include "books.h"
#define taskname "test"
#define fi first
#define se second
#define pb push_back
#define faster ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
using ll = long long;
using pii = pair <int, int>;
using pil = pair <int, ll>;
using pli = pair <ll, int>;
using pll = pair <ll, ll>;
using ull = unsigned ll;

ll min(const ll &a, const ll &b){
    return (a < b) ? a : b;
}

ll max(const ll &a, const ll &b){
    return (a > b) ? a : b;
}

//const ll Mod = 1000000009;
//const ll Mod2 = 999999999989;
//only use when required
const int maxN = 1e5 + 1;

ll n, k, a, s;
vector <pll> tem;
vector <int> ans;

void solve(int N, int K, ll A, int S){
    n = N; k = K; a = A; s = S;
    ll sum = 0;
    for (int i = 1; i <= k; ++i){
        tem.pb({i, skim(i)});
        sum += tem.back().se;
    }
    if (sum > 2 * a) impossible();
    if (a <= sum && sum <= 2 * a){
        for (auto i: tem) ans.pb(i.fi);
        answer(ans);
    }
    int l = 1, r = n;
    while (l <= r){
        int mid = (l + r) / 2;
        if (skim(mid) >= a) r = mid - 1;
        else l = mid + 1;
    }
    if (r <= k) impossible();
    ++r;
    if (r <= n){
        sum = skim(r);
        for (int i = 0; i < k - 1; ++i) sum += tem[i].se;
        if (sum <= 2 * a){
            for (int i = 1; i < k; ++i) ans.pb(i);
            ans.pb(r);
            answer(ans);
        }
    }
    --r;
    if (r == k) impossible();
    sum = 0;
    ll atr = skim(r);
    for (int i = 1; i < k; ++i) sum += tem[i].se;
    if (sum + atr >= a){
        for (int i = 2; i <= k; ++i) ans.pb(i);
        ans.pb(r);
        answer(ans);
    }
    tem.pb({r, atr});
    sum = atr;
    ans.pb(r);
    for (int i = r - 1; i > max(k, r - k) && sum < a; --i){
        tem.pb({i, skim(i)});
        ans.pb(i);
        sum += tem.back().se;
    }
    if (sum < a) impossible();
    for (int i = 1; i <= k - ans.size(); ++i){
        ans.pb(i);
    }
    answer(ans);
}

Compilation message (stderr)

books.cpp: In function 'void solve(int, int, ll, int)':
books.cpp:81:23: warning: comparison of integer expressions of different signedness: 'int' and 'long long unsigned int' [-Wsign-compare]
   81 |     for (int i = 1; i <= k - ans.size(); ++i){
      |                     ~~^~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 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