This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <map>
#include "books.h"
using namespace std;
map<int, long long> m;
int s;
long long ask(int x) {
if (s) {
s--;
return skim(x);
} else return (long long)-1;
}
long long get(int x, int K) {
long long ret = 0;
for (int i = 0; i < K; i++) {
if (m.find(x + i + 1) == m.end()) m[x + i + 1] = ask(x + i + 1);
ret += m[x + i + 1];
if (ret > (long long)4e18) ret = (long long)4e18;
}
return ret;
}
void solve(int N, int K, long long A, int S) {
s = S;
int l = 0, r = N - K;
while (l < r) {
int m = (l + r) / 2;
long long t = get(m, 1);
if (t < A / K) l = m + 1;
else r = m;
}
long long sum = 0;
for (int i = l; i < l + K; i++) {
sum += get(i, 1);
if (sum >= A) {
vector<int> v;
for (int j = 0; j < K; j++) v.push_back(i - K + j + 2);
answer(v);
}
}
if (l >= N) {
impossible();
return;
}
r = 0;
while (r + K - 1 <= l && r <= 0) {
if (get(r, K - 1) + get(l, 1) > 2 * A) break;
if (get(r, K - 1) + get(l, 1) >= A || s <= 0) {
vector<int> v;
for (int i = 0; i < K - 1; i++) v.push_back(r + i + 1);
v.push_back(l + 1);
answer(v);
return;
}
r++;
}
impossible();
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |