#include "prize.h" #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #define FOR(i, x, y) for (int i = x; i < y; i++) typedef long long ll; using namespace std; using namespace __gnu_pbds; typedef tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update> indexed_set; int find_best(int n) { int mx = 0, worst = -1; vector<int> worst_res; // Finds the first lollipop FOR(i, 0, 474) { vector<int> res = ask(i); if (res[0] + res[1] >= 22) { worst = i; worst_res = res; break; } else if (res[0] + res[1] > mx) { mx = res[0] + res[1]; worst = i; worst_res = res; } else if (res[0] + res[1] == 0) return i; } // Binary searches for the other prizes indexed_set remaining; FOR(i, 474, n) remaining.insert(i); while (remaining.size()) { int l = 0, r = remaining.size() - 1; while (l != r) { int mid = (l + r + 1) / 2; int pos = *remaining.find_by_order(mid); vector<int> res = ask(pos); if (res[0] + res[1] == 0) return pos; else if (res[0] + res[1] == worst_res[0] + worst_res[1]) { if (res[1] == worst_res[1]) { while (*remaining.begin() != pos) remaining.erase(remaining.begin()); remaining.erase(remaining.begin()); break; } else r = mid - 1; } else { remaining.erase(pos); break; } } } }

prize.cpp: In function 'int find_best(int)':
prize.cpp:12:14: warning: variable 'worst' set but not used [-Wunused-but-set-variable]
  int mx = 0, worst = -1;
              ^~~~~
prize.cpp:50:1: warning: control reaches end of non-void function [-Wreturn-type]
 }
 ^