# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
401155 | galca | A Difficult(y) Choice (BOI21_books) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
using namespace std;
int findA(vector<long long>& books, long long A, int lower, int upper) {
if (books[lower] == -1) {
books[lower] = skim(lower + 1);
}
if (books[upper] == -1) {
books[upper] = skim(upper + 1);
}
if (books[lower] > A) {
return -1;
}
if (books[upper] <= A) {
return upper;
}
if (upper - lower < 2) {
return lower;
}
int half = (upper - lower + 1) / 2;
books[lower + half] = skim(lower + half + 1);
if (books[lower + half] < A) return findA(books, A, lower + half, upper);
else return findA(books, A, lower, lower + half);
}
void solve(int N, int K, long long A, int S) {
if (K > N) {
impossible();
return;
}
vector<long long> books(N);
for (int i = 0; i < N; i++) {
books[i] = -1;
}
int posA = findA(books, A, 0, N - 1);
if (posA == -1) {
impossible();
}
else {
if (posA == (N - 1)) {
long long sum = 0;
vector<int> res;
for (int i = 0; i < K; i++) {
res.push_back(N - i);
if (books[N - i - 1] == -1) {
books[N - i - 1] = skim(N - i);
}
sum += books[N - i - 1];
}
if (sum < A) {
impossible();
}
answer(res);
}
else {
vector<int> res;
int lastPos = posA + 1 + K;
if (lastPos > N) {
lastPos = N;
}
for (int i = 0; i < K; i++) {
res.push_back(lastPos - i);
}
answer(res);
}
}
return;
}