# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
708156 | Cyanmond | 커다란 상품 (IOI17_prize) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "prize.h"
#include <bits/stdc++.h>
std::map<int, std::vector<int>> askMemo;
std::vector<int> Ask(int i) {
if (askMemo.find(i) != askMemo.end()) return askMemo[i];
return askMemo[i] = ask(i);
}
std::vector<int> findLower(std::vector<int> ids) {
auto question = [&](const int i) {
return Ask(ids[i]);
};
const int n = (int)ids.size();
const int limit = lm(n);
std::vector<std::vector<int>> vals;
for (int i = 0; i <= limit; ++i) {
vals.push_back(question(i));
}
int df = 0;
for (const auto &vec : vals) {
df = std::max(df, vec[0] + vec[1]);
}
std::vector<int> lws;
auto relax = [&](int i, std::vector<int> &ans) {
for (const auto e : lws) {
if (e < i) --ans[0];
if (e > i) --ans[1];
}
};
auto solve = [&](auto &&self, const int l, const int r, const int lv, const int rv) -> void {
const int mid = (l + r) / 2;
int lm = -1;
for (int i = mid; i < r; ++i) {
auto x = question(i);
if (x[0] + x[1] != df) {
lws.push_back(ids[i]);
} else {
lm = i;
break;
}
}
int rm = n + 1;
for (int i = mid; i >= l; --i) {
auto x = question(i);
if (x[0] + x[1] != df) {
lws.push_back(ids[i]);
} else {
rm = i;
break;
}
}
if (rm != n + 1) {
auto x = question(rm);
// relax(rm, x);
const int newLv = lv, newRv = x[1];
if (newLv + newRv != df) {
self(self, l, rm, newLv, newRv);
}
}
if (lm != -1) {
auto x = question(lm);
const int newLv = x[0], newRv = rv;
if (newLv + newRv != df) {
self(self, lm + 1, r, newLv, newRv);
}
}
};
solve(solve, 0, n, 0, 0);
std::sort(lws.begin(), lws.end());
return lws;
}
int find_best(int n) {
std::vector<int> ids(n);
std::iota(ids.begin(), ids.end(), 0);
while (ids.size() != 1) {
ids = findLower(ids);
}
return ids[0];
}