#include "books.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int n, k, s;
ll a;
void solve(int N, int K, ll A, int S) {
n = N;
k = K;
a = A;
s = S;
vector<ll> arr(n+1);
for (int i = 1; i <= n; i++) {
skim(arr[i]);
}
ll currS = 0;
vector<int> index(k+2);
for (int i = 1; i <= k; i++) {
currS += arr[i];
index[i] = i;
}
index[k+1] = n+1;
for (int curr = k; curr >= 1; curr--) {
int allow = index[curr+1] - 1;
currS -= arr[curr];
for (int j = curr+1; j <= allow; j++) {
currS += arr[j];
index[curr] = j;
if (currS >= a && currS <= 2*a) {
vector<int> fin;
for (int i = 1; i <= k; i++) {
fin.push_back(index[i]);
}
answer(fin);
return;
}
else if (currS < a) {
if (j == allow) break;
currS -= arr[j];
continue;
}
else if (currS > 2*a) {
currS -= arr[j];
index[curr] = j-1;
currS += arr[j-1];
break;
}
}
}
if (currS > 2 * a || currS < a) {
impossible();
return;
}
else {
vector<int> fin;
for (int i = 1; i <= k; i++) {
fin.push_back(index[i]);
}
answer(fin);
return;
}
}
# | 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... |