#include "prize.h"
#include<bits/stdc++.h>
using namespace std;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int B = 600;
const int maxn = 2e5+5;
vector<int> res[maxn];
vector<int> get(int x){
if(!res[x].empty()) return res[x];
res[x]=ask(x);res[x][1]+=res[x][0];
return res[x];
}
int N,bit[maxn],Max;
void update(int x){
for(int i=x+1;i<=N;i+=(i&(-i))) bit[i]++;
}
int query(int x){
int ans=0;
for(int i=x+1;i>=1;i-=(i&(-i))) ans+=bit[i];
return ans;
}
int find_best(int n){
N=n;
vector<int> ord(min(n,500));
iota(ord.begin(),ord.end(),0);
shuffle(ord.begin(),ord.end(),rng);
for(int i=0;i<500;i++){
vector<int> cc=get(i);
if(cc[1]==0) return i;
Max=max(Max,cc[1]);
}
for(int i=0;i<n;i+=B){
vector<int> p;
for(int j=i;j<min(i+B,n);j++) p.push_back(j);
int cnt=0;
while(!p.empty()){
int x=p.back();p.pop_back();
vector<int> cc=get(x);
if(cc[1]==0) return x;
if(cc[1]<Max) update(x);
else{
cnt=cc[0]-query(x);
break;
}
}
while(cnt){
int l=0,r=(int)p.size()-1;
bool check=false;
while(l<=r){
int mid=(l+r)>>1;
vector<int> cc=get(p[mid]);
if(cc[1]==0) return p[mid];
if(cc[1]<Max){
update(p[mid]);
p.erase(p.begin()+mid);
check=true;
break;
}
if(cc[0]-query(p[mid])) r=mid-1;
else l=mid+1;
}
cnt--;
}
}
return -1;
}
Compilation message
prize.cpp: In function 'int find_best(int)':
prize.cpp:52:18: warning: variable 'check' set but not used [-Wunused-but-set-variable]
52 | bool check=false;
| ^~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
6012 KB |
Output is correct |
2 |
Correct |
7 ms |
5984 KB |
Output is correct |
3 |
Correct |
4 ms |
6012 KB |
Output is correct |
4 |
Correct |
5 ms |
6028 KB |
Output is correct |
5 |
Correct |
7 ms |
6524 KB |
Output is correct |
6 |
Correct |
1 ms |
5720 KB |
Output is correct |
7 |
Correct |
4 ms |
5784 KB |
Output is correct |
8 |
Correct |
4 ms |
6012 KB |
Output is correct |
9 |
Correct |
5 ms |
6032 KB |
Output is correct |
10 |
Correct |
8 ms |
5756 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
6016 KB |
Output is correct |
2 |
Correct |
6 ms |
6260 KB |
Output is correct |
3 |
Correct |
4 ms |
6284 KB |
Output is correct |
4 |
Correct |
5 ms |
5784 KB |
Output is correct |
5 |
Correct |
4 ms |
6012 KB |
Output is correct |
6 |
Correct |
1 ms |
5716 KB |
Output is correct |
7 |
Correct |
6 ms |
6024 KB |
Output is correct |
8 |
Correct |
5 ms |
6268 KB |
Output is correct |
9 |
Correct |
4 ms |
6036 KB |
Output is correct |
10 |
Correct |
5 ms |
6036 KB |
Output is correct |
11 |
Correct |
6 ms |
5760 KB |
Output is correct |
12 |
Correct |
3 ms |
6020 KB |
Output is correct |
13 |
Correct |
6 ms |
5756 KB |
Output is correct |
14 |
Correct |
4 ms |
6004 KB |
Output is correct |
15 |
Correct |
6 ms |
6160 KB |
Output is correct |
16 |
Correct |
22 ms |
6380 KB |
Output is correct |
17 |
Correct |
4 ms |
6016 KB |
Output is correct |
18 |
Correct |
29 ms |
6944 KB |
Output is correct |
19 |
Correct |
4 ms |
6016 KB |
Output is correct |
20 |
Correct |
8 ms |
5892 KB |
Output is correct |
21 |
Correct |
9 ms |
6868 KB |
Output is correct |
22 |
Correct |
8 ms |
5988 KB |
Output is correct |
23 |
Correct |
3 ms |
5764 KB |
Output is correct |
24 |
Correct |
4 ms |
5760 KB |
Output is correct |
25 |
Correct |
13 ms |
6508 KB |
Output is correct |
26 |
Correct |
14 ms |
6304 KB |
Output is correct |
27 |
Correct |
5 ms |
6032 KB |
Output is correct |
28 |
Correct |
24 ms |
6540 KB |
Output is correct |
29 |
Correct |
16 ms |
6476 KB |
Output is correct |
30 |
Correct |
28 ms |
6968 KB |
Output is correct |
31 |
Correct |
3 ms |
5756 KB |
Output is correct |
32 |
Correct |
6 ms |
6280 KB |
Output is correct |
33 |
Correct |
2 ms |
5720 KB |
Output is correct |
34 |
Correct |
9 ms |
6320 KB |
Output is correct |
35 |
Correct |
5 ms |
6028 KB |
Output is correct |
36 |
Correct |
8 ms |
6676 KB |
Output is correct |
37 |
Correct |
5 ms |
5804 KB |
Output is correct |
38 |
Correct |
4 ms |
6008 KB |
Output is correct |
39 |
Correct |
12 ms |
6220 KB |
Output is correct |
40 |
Correct |
23 ms |
7168 KB |
Output is correct |
41 |
Correct |
13 ms |
6052 KB |
Output is correct |
42 |
Correct |
13 ms |
6716 KB |
Output is correct |
43 |
Correct |
13 ms |
6428 KB |
Output is correct |
44 |
Correct |
10 ms |
6212 KB |
Output is correct |
45 |
Correct |
11 ms |
6312 KB |
Output is correct |
46 |
Correct |
3 ms |
6004 KB |
Output is correct |
47 |
Correct |
10 ms |
6484 KB |
Output is correct |
48 |
Correct |
16 ms |
6508 KB |
Output is correct |
49 |
Correct |
6 ms |
6564 KB |
Output is correct |
50 |
Correct |
29 ms |
6484 KB |
Output is correct |
51 |
Correct |
9 ms |
6680 KB |
Output is correct |
52 |
Correct |
4 ms |
6264 KB |
Output is correct |
53 |
Correct |
5 ms |
5776 KB |
Output is correct |
54 |
Correct |
12 ms |
6396 KB |
Output is correct |
55 |
Correct |
3 ms |
6016 KB |
Output is correct |
56 |
Correct |
25 ms |
6756 KB |
Output is correct |
57 |
Correct |
19 ms |
6900 KB |
Output is correct |
58 |
Correct |
16 ms |
6520 KB |
Output is correct |
59 |
Correct |
14 ms |
6300 KB |
Output is correct |
60 |
Correct |
15 ms |
6504 KB |
Output is correct |
61 |
Correct |
5 ms |
6288 KB |
Output is correct |
62 |
Correct |
4 ms |
6008 KB |
Output is correct |
63 |
Correct |
5 ms |
6024 KB |
Output is correct |
64 |
Correct |
5 ms |
6288 KB |
Output is correct |
65 |
Correct |
5 ms |
6052 KB |
Output is correct |
66 |
Correct |
6 ms |
6040 KB |
Output is correct |
67 |
Correct |
7 ms |
6288 KB |
Output is correct |
68 |
Correct |
4 ms |
6012 KB |
Output is correct |
69 |
Correct |
7 ms |
6284 KB |
Output is correct |
70 |
Correct |
7 ms |
6276 KB |
Output is correct |
71 |
Partially correct |
34 ms |
6272 KB |
Partially correct - number of queries: 5413 |
72 |
Correct |
6 ms |
6292 KB |
Output is correct |
73 |
Partially correct |
25 ms |
6796 KB |
Partially correct - number of queries: 5350 |
74 |
Partially correct |
29 ms |
6396 KB |
Partially correct - number of queries: 5377 |
75 |
Correct |
6 ms |
6044 KB |
Output is correct |
76 |
Correct |
20 ms |
6556 KB |
Output is correct |
77 |
Correct |
24 ms |
6344 KB |
Output is correct |
78 |
Correct |
7 ms |
6516 KB |
Output is correct |
79 |
Correct |
15 ms |
6768 KB |
Output is correct |
80 |
Correct |
26 ms |
6528 KB |
Output is correct |
81 |
Correct |
19 ms |
6520 KB |
Output is correct |
82 |
Correct |
24 ms |
6820 KB |
Output is correct |
83 |
Correct |
6 ms |
6008 KB |
Output is correct |
84 |
Correct |
18 ms |
6776 KB |
Output is correct |
85 |
Correct |
27 ms |
6600 KB |
Output is correct |
86 |
Correct |
6 ms |
6160 KB |
Output is correct |
87 |
Correct |
5 ms |
5756 KB |
Output is correct |
88 |
Correct |
6 ms |
6008 KB |
Output is correct |
89 |
Correct |
5 ms |
6260 KB |
Output is correct |
90 |
Correct |
5 ms |
6516 KB |
Output is correct |
91 |
Correct |
6 ms |
6004 KB |
Output is correct |
92 |
Correct |
5 ms |
6012 KB |
Output is correct |
93 |
Correct |
6 ms |
5856 KB |
Output is correct |
94 |
Correct |
6 ms |
6048 KB |
Output is correct |
95 |
Correct |
7 ms |
6828 KB |
Output is correct |
96 |
Correct |
7 ms |
6092 KB |
Output is correct |
97 |
Correct |
6 ms |
5800 KB |
Output is correct |