#include "prize.h"
#include <iostream>
#include <vector>
#include <random>
#include <ctime>
using namespace std;
const int maxn = 200005;
int N;
int max_result = 0;
int divide(int l, int r, int removed_from_left, int removed_from_right)
{
if (l > r)
{
return 0;
}
int mid = (l + r) / 2;
vector<int> curr;
while (mid <= r)
{
curr = ask(mid);
if (curr[0] + curr[1] == 0)
{
return mid;
}
if (curr[0] + curr[1] == max_result)
{
break;
}
++mid;
}
if (mid <= r)
{
curr[0] -= (removed_from_left - (mid - (l + r) / 2));
curr[1] -= removed_from_right;
if (curr[0] > 0)
{
int idx = divide(l, (l + r) / 2 - 1, removed_from_left, removed_from_right + curr[1] + (mid - (l + r) / 2));
if (idx != 0)
{
return idx;
}
}
if (curr[1] > 0)
{
int idx = divide(mid + 1, r, removed_from_left + curr[0] + (mid - (l + r) / 2), removed_from_right);
if (idx != 0)
{
return idx;
}
}
return 0;
}
else
{
mid = (l + r) / 2 - 1;
while (mid >= l)
{
curr = ask(mid);
if (curr[0] + curr[1] == 0)
{
return mid;
}
if (curr[0] + curr[1] == max_result)
{
return mid;
}
--mid;
}
if (mid == l - 1)
{
return 0;
}
curr[0] -= removed_from_left;
if (curr[0] > 0)
{
return divide(l, mid - 1, removed_from_left, removed_from_right + (r - mid));
}
return 0;
}
return 0;
}
bool asked[maxn];
int find_best(int n)
{
srand(time(NULL));
N = n;
for (int i = 1; i <= 465; )
{
int idx = rand() % N;
if (asked[idx] == true)
{
continue;
}
asked[idx] = true;
vector<int> curr_res = ask(idx);
if (curr_res[0] + curr_res[1] == 0)
{
return idx;
}
max_result = max(max_result, curr_res[0] + curr_res[1]);
}
return divide(0, N - 1, 0, 0);
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
81 ms |
464 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
112 ms |
596 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |