#include "prize.h"
#include <bits/stdc++.h>
using namespace std;
#define pii pair<int, int>
#define mp make_pair
#define maxn 200010
int blk;
int qs[maxn][2];
int numqueries = 0;
pii query(int spot) {
if (qs[spot][0] != -1)
return pii(qs[spot][0], qs[spot][1]);
numqueries++;
assert(numqueries <= 10000); //induce run time error instead of WA
//seems that above got hit
vector<int> res = ask(spot);
qs[spot][0] = res[0];
qs[spot][1] = res[1];
return pii(res[0], res[1]);
}
queue<pair<pii, pii>> q;
int find_best(int n) {
//we don't really care what happens for smaller n,
// b/c it is not a function of n
for (int i = 0; i < n; i++) {
qs[i][0] = qs[i][1] = -1;
}
//i am a little worried about exact blk bound
blk = sqrt(n)+50; //changed from 5 to 50
blk = min(blk, n); //just making sure
int mxind = -1;
int mxstuff = -1;
for(int i = 0; i < blk; i++) {
pii res = query(i);
if(res.first + res.second == 0)
return i; //this will always be there
if (res.first+ res.second >= mxstuff) {
mxstuff = res.first + res.second;
mxind = i;
}
}
int aft = query(mxind).second;
int numtot = mxstuff; //total number less than biggest guy
int nc = 0;
for (int i = 0; i < blk; i++) {
pii cur = query(i);
if (cur.first + cur.second != numtot) nc++;
}
q.push(mp(mp(blk, n-1), mp(nc, 0)));
//gives number to left & right of range
while (q.size()) {
auto cur = q.front(); q.pop();
int cleft = cur.first.first;
int cright = cur.first.second;
if (cleft > cright) continue; //why would we care
int numtoleft = cur.second.first;
int numtoright = cur.second.second;
if (numtoleft + numtoright == numtot) continue; //why bother
if (cleft == cright) {
pii cur = query(cleft);
if (cur.first + cur.second == 0) return cleft;
continue;
}
bool done = false;
int mid = (cleft + cright)/2;
int omid = mid;
int tookout = 0;
while (mid != cright+1) {
pii cur = query(mid);
if (cur.first + cur.second == numtot) {
q.push(mp(mp(cleft, omid-1),
mp(numtoleft, cur.second + tookout)));
q.push(mp(mp(mid+1, cright),
mp(cur.first, numtoright)));
done = true;
break;
}
else {
tookout++;
if (cur.first + cur.second == 0) return mid;
mid++;
}
}
if (done) continue;
mid = omid-1;
while (mid != cleft-1) {
pii cur = query(mid);
if (cur.first + cur.second == numtot) {
q.push(mp(mp(cleft , mid-1),
mp(numtoleft, cur.second)));
done = true;
break;
}
else {
if (cur.first + cur.second == 0) return mid;
mid--;
}
}
}
return 0;
}
//at most sqrt(n) guys that are not the maximums
//if we query smth not the maximum, we split it in half
//grader can behave adversarily
Compilation message
prize.cpp: In function 'int find_best(int)':
prize.cpp:49:6: warning: unused variable 'aft' [-Wunused-variable]
int aft = query(mxind).second;
^~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
6 ms |
1912 KB |
Output is correct |
| 2 |
Correct |
7 ms |
1988 KB |
Output is correct |
| 3 |
Correct |
7 ms |
1988 KB |
Output is correct |
| 4 |
Correct |
8 ms |
1988 KB |
Output is correct |
| 5 |
Correct |
5 ms |
2088 KB |
Output is correct |
| 6 |
Correct |
3 ms |
2088 KB |
Output is correct |
| 7 |
Correct |
8 ms |
2088 KB |
Output is correct |
| 8 |
Correct |
7 ms |
2088 KB |
Output is correct |
| 9 |
Correct |
6 ms |
2088 KB |
Output is correct |
| 10 |
Correct |
6 ms |
2088 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
14 ms |
2088 KB |
Output is correct |
| 2 |
Correct |
6 ms |
2088 KB |
Output is correct |
| 3 |
Correct |
8 ms |
2100 KB |
Output is correct |
| 4 |
Correct |
5 ms |
2108 KB |
Output is correct |
| 5 |
Correct |
7 ms |
2108 KB |
Output is correct |
| 6 |
Correct |
3 ms |
2108 KB |
Output is correct |
| 7 |
Correct |
7 ms |
2108 KB |
Output is correct |
| 8 |
Correct |
7 ms |
2108 KB |
Output is correct |
| 9 |
Correct |
6 ms |
2108 KB |
Output is correct |
| 10 |
Correct |
8 ms |
2108 KB |
Output is correct |
| 11 |
Correct |
9 ms |
2108 KB |
Output is correct |
| 12 |
Correct |
8 ms |
2112 KB |
Output is correct |
| 13 |
Correct |
7 ms |
2140 KB |
Output is correct |
| 14 |
Correct |
10 ms |
2140 KB |
Output is correct |
| 15 |
Correct |
21 ms |
2140 KB |
Output is correct |
| 16 |
Correct |
22 ms |
2140 KB |
Output is correct |
| 17 |
Correct |
6 ms |
2140 KB |
Output is correct |
| 18 |
Correct |
29 ms |
2236 KB |
Output is correct |
| 19 |
Correct |
6 ms |
2236 KB |
Output is correct |
| 20 |
Correct |
21 ms |
2236 KB |
Output is correct |
| 21 |
Correct |
31 ms |
2236 KB |
Output is correct |
| 22 |
Correct |
25 ms |
2236 KB |
Output is correct |
| 23 |
Correct |
7 ms |
2236 KB |
Output is correct |
| 24 |
Correct |
8 ms |
2236 KB |
Output is correct |
| 25 |
Correct |
19 ms |
2236 KB |
Output is correct |
| 26 |
Correct |
30 ms |
2236 KB |
Output is correct |
| 27 |
Correct |
5 ms |
2236 KB |
Output is correct |
| 28 |
Correct |
21 ms |
2236 KB |
Output is correct |
| 29 |
Correct |
27 ms |
2236 KB |
Output is correct |
| 30 |
Correct |
23 ms |
2236 KB |
Output is correct |
| 31 |
Correct |
5 ms |
2236 KB |
Output is correct |
| 32 |
Correct |
7 ms |
2236 KB |
Output is correct |
| 33 |
Correct |
2 ms |
2236 KB |
Output is correct |
| 34 |
Correct |
31 ms |
2236 KB |
Output is correct |
| 35 |
Correct |
8 ms |
2236 KB |
Output is correct |
| 36 |
Correct |
30 ms |
2236 KB |
Output is correct |
| 37 |
Correct |
8 ms |
2236 KB |
Output is correct |
| 38 |
Correct |
9 ms |
2236 KB |
Output is correct |
| 39 |
Correct |
23 ms |
2236 KB |
Output is correct |
| 40 |
Correct |
28 ms |
2236 KB |
Output is correct |
| 41 |
Correct |
35 ms |
2236 KB |
Output is correct |
| 42 |
Correct |
22 ms |
2236 KB |
Output is correct |
| 43 |
Correct |
33 ms |
2236 KB |
Output is correct |
| 44 |
Correct |
22 ms |
2236 KB |
Output is correct |
| 45 |
Correct |
24 ms |
2236 KB |
Output is correct |
| 46 |
Correct |
6 ms |
2236 KB |
Output is correct |
| 47 |
Correct |
25 ms |
2236 KB |
Output is correct |
| 48 |
Correct |
41 ms |
2236 KB |
Output is correct |
| 49 |
Correct |
34 ms |
2236 KB |
Output is correct |
| 50 |
Correct |
32 ms |
2236 KB |
Output is correct |
| 51 |
Correct |
33 ms |
2236 KB |
Output is correct |
| 52 |
Correct |
44 ms |
2236 KB |
Output is correct |
| 53 |
Correct |
8 ms |
2236 KB |
Output is correct |
| 54 |
Correct |
24 ms |
2236 KB |
Output is correct |
| 55 |
Correct |
8 ms |
2236 KB |
Output is correct |
| 56 |
Correct |
40 ms |
2252 KB |
Output is correct |
| 57 |
Correct |
51 ms |
2252 KB |
Output is correct |
| 58 |
Correct |
33 ms |
2252 KB |
Output is correct |
| 59 |
Correct |
14 ms |
2252 KB |
Output is correct |
| 60 |
Correct |
34 ms |
2252 KB |
Output is correct |
| 61 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 62 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 63 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 64 |
Correct |
5 ms |
2252 KB |
Output is correct |
| 65 |
Correct |
5 ms |
2252 KB |
Output is correct |
| 66 |
Correct |
7 ms |
2252 KB |
Output is correct |
| 67 |
Correct |
7 ms |
2252 KB |
Output is correct |
| 68 |
Correct |
5 ms |
2252 KB |
Output is correct |
| 69 |
Correct |
10 ms |
2252 KB |
Output is correct |
| 70 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 71 |
Partially correct |
38 ms |
2252 KB |
Partially correct - number of queries: 5256 |
| 72 |
Correct |
10 ms |
2252 KB |
Output is correct |
| 73 |
Partially correct |
35 ms |
2252 KB |
Partially correct - number of queries: 5193 |
| 74 |
Partially correct |
29 ms |
2252 KB |
Partially correct - number of queries: 5220 |
| 75 |
Correct |
8 ms |
2252 KB |
Output is correct |
| 76 |
Correct |
22 ms |
2252 KB |
Output is correct |
| 77 |
Correct |
35 ms |
2252 KB |
Output is correct |
| 78 |
Correct |
13 ms |
2252 KB |
Output is correct |
| 79 |
Correct |
17 ms |
2252 KB |
Output is correct |
| 80 |
Correct |
22 ms |
2252 KB |
Output is correct |
| 81 |
Correct |
35 ms |
2252 KB |
Output is correct |
| 82 |
Correct |
22 ms |
2252 KB |
Output is correct |
| 83 |
Correct |
7 ms |
2252 KB |
Output is correct |
| 84 |
Correct |
30 ms |
2252 KB |
Output is correct |
| 85 |
Correct |
34 ms |
2252 KB |
Output is correct |
| 86 |
Correct |
6 ms |
2252 KB |
Output is correct |
| 87 |
Correct |
5 ms |
2252 KB |
Output is correct |
| 88 |
Correct |
7 ms |
2252 KB |
Output is correct |
| 89 |
Correct |
5 ms |
2252 KB |
Output is correct |
| 90 |
Correct |
8 ms |
2252 KB |
Output is correct |
| 91 |
Correct |
8 ms |
2252 KB |
Output is correct |
| 92 |
Correct |
7 ms |
2252 KB |
Output is correct |
| 93 |
Correct |
10 ms |
2252 KB |
Output is correct |
| 94 |
Correct |
10 ms |
2252 KB |
Output is correct |
| 95 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 96 |
Correct |
9 ms |
2252 KB |
Output is correct |
| 97 |
Correct |
6 ms |
2252 KB |
Output is correct |
