| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 399532 | Temmie | A Difficult(y) Choice (BOI21_books) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "books.h"
#include <bits/stdc++.h>
typedef long long ll;
void solve(int n, int k, int s, ll a) {
std::vector <ll> vals(n + 1, -1);
ll sumfirst = 0;
for (int i = 1; i <= k - 1; i++) {
vals[i] = ask(i);
sumfirst += vals[i];
}
int l = k + 1, r = n;
int best = k;
while (l <= r) {
int mid = (l + r) >> 1;
ll now = ask(mid);
vals[mid] = now;
ll total = now + sumfirst;
if (total >= a + a) {
r = mid - 1;
} else {
l = mid + 1;
best = mid;
}
}
if (best == k && sumfirst + vals[best] >= a + a) {
impossible();
return;
}
if (sumfirst + vals[best] >= a) {
std::vector <int> ans;
for (int i = 1; i <= k - 1; i++) ans.push_back(i);
ans.push_back(best);
answer(ans);
return;
}
std::vector <int> ans;
for (int i = 1; i <= k - 1; i++) ans.push_back(i);
ans.push_back(best);
std::set <int> have;
for (int x : ans) have.insert(x);
ll sum = sumfirst + vals[best];
int cur = best;
for (int i = 1 - 1; i <= k - 1 - 1; i++) {
sum -= vals[ans[i]];
have.erase(ans[i]);
int nw = --cur;
if (have.count(nw)) {
impossible();
return;
}
have.insert(nw);
if (vals[nw] == -1LL) vals[nw] = ask(nw);
sum += vals[nw];
if (sum >= a) break;
}
std::vector <int> put;
for (int x : have) put.push_back(x);
answer(put);
}