#include <bits/stdc++.h>
#include "insects.h"
//#include "stub.cpp"
using namespace std;
int l,r,mid;
vector <int> PrevIndexIn,PrevIndexOut,RemainingInsects;
int InMachine[2003];
int n;
int TypeCount=0;
int CurrentInMachine=0;
int PossibleMincount(int cnt){
//Pushing insects inside the machine, keeping track of the ones being putted inside for this iteration of binary searching
mt19937 seed(63222221001217544);
shuffle(RemainingInsects.begin(),RemainingInsects.end(),seed);
for(int i=0;i<RemainingInsects.size();i++){
if(CurrentInMachine==cnt*TypeCount){
PrevIndexOut.push_back(RemainingInsects[i]);
continue;
}
move_inside(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=1;
PrevIndexIn.push_back(RemainingInsects[i]);
CurrentInMachine++;
if(press_button()>cnt){
PrevIndexIn.pop_back();
PrevIndexOut.push_back(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=0;
move_outside(RemainingInsects[i]);
CurrentInMachine--;
}
}
RemainingInsects.clear();
if(CurrentInMachine<cnt*TypeCount){
//There clearly exist a type of insect with cardinality less than cnt, so the answer is less than cnt. Removing insects from last iteration.
while(!PrevIndexIn.empty()){
InMachine[PrevIndexIn.back()]=0;
move_outside(PrevIndexIn.back());
PrevIndexOut.push_back(PrevIndexIn.back());
PrevIndexIn.pop_back();
CurrentInMachine--;
}
RemainingInsects=PrevIndexOut;
PrevIndexOut.clear();
return 0;
}else if(CurrentInMachine==cnt*TypeCount){
//As every type of insects has at least cnt insects, the answer is at least cnt. Just return true
RemainingInsects=PrevIndexOut;
PrevIndexIn.clear();
PrevIndexOut.clear();
return 1;
}
}
int min_cardinality(int N){
//Remember to initialize as this is multitest
for(int i=0;i<N;i++){
InMachine[i]=0;
}
PrevIndexIn.clear();
PrevIndexOut.clear();
RemainingInsects.clear();
n=N;
l=1;
//Phase 1: Determining the amount of different types of insects
TypeCount=0;
CurrentInMachine=0;
for(int i=0;i<n;i++){
move_inside(i);
InMachine[i]=1;
CurrentInMachine++;
if(press_button()>1){
move_outside(i);
InMachine[i]=0;
CurrentInMachine--;
}else{
TypeCount++;
}
}
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
r=(RemainingInsects.size()/TypeCount+1);
int ans=1;
//Phase 2: Binary search for the answer
while(r-l>=1){
mid=(l+r+1)/2;
if(PossibleMincount(mid)){
ans=max(ans,mid);
l=mid;
r=min(r,mid+(n-CurrentInMachine)/TypeCount);
}else{
r=mid-1;
}
}
return ans;
}
Compilation message
insects.cpp: In function 'int PossibleMincount(int)':
insects.cpp:16:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
16 | for(int i=0;i<RemainingInsects.size();i++){
| ~^~~~~~~~~~~~~~~~~~~~~~~~
insects.cpp:53:1: warning: control reaches end of non-void function [-Wreturn-type]
53 | }
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
344 KB |
Output is correct |
4 |
Correct |
0 ms |
344 KB |
Output is correct |
5 |
Correct |
0 ms |
344 KB |
Output is correct |
6 |
Correct |
2 ms |
600 KB |
Output is correct |
7 |
Correct |
1 ms |
344 KB |
Output is correct |
8 |
Correct |
2 ms |
344 KB |
Output is correct |
9 |
Correct |
3 ms |
600 KB |
Output is correct |
10 |
Correct |
7 ms |
440 KB |
Output is correct |
11 |
Correct |
1 ms |
344 KB |
Output is correct |
12 |
Correct |
5 ms |
688 KB |
Output is correct |
13 |
Correct |
3 ms |
344 KB |
Output is correct |
14 |
Correct |
2 ms |
344 KB |
Output is correct |
15 |
Correct |
2 ms |
344 KB |
Output is correct |
16 |
Correct |
2 ms |
600 KB |
Output is correct |
17 |
Correct |
3 ms |
600 KB |
Output is correct |
18 |
Correct |
3 ms |
600 KB |
Output is correct |
19 |
Correct |
2 ms |
440 KB |
Output is correct |
20 |
Correct |
2 ms |
344 KB |
Output is correct |
21 |
Correct |
2 ms |
344 KB |
Output is correct |
22 |
Correct |
1 ms |
344 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
344 KB |
Output is correct |
4 |
Correct |
0 ms |
344 KB |
Output is correct |
5 |
Correct |
0 ms |
344 KB |
Output is correct |
6 |
Correct |
2 ms |
600 KB |
Output is correct |
7 |
Correct |
1 ms |
344 KB |
Output is correct |
8 |
Correct |
2 ms |
344 KB |
Output is correct |
9 |
Correct |
3 ms |
600 KB |
Output is correct |
10 |
Correct |
7 ms |
440 KB |
Output is correct |
11 |
Correct |
1 ms |
344 KB |
Output is correct |
12 |
Correct |
5 ms |
688 KB |
Output is correct |
13 |
Correct |
3 ms |
344 KB |
Output is correct |
14 |
Correct |
2 ms |
344 KB |
Output is correct |
15 |
Correct |
2 ms |
344 KB |
Output is correct |
16 |
Correct |
2 ms |
600 KB |
Output is correct |
17 |
Correct |
3 ms |
600 KB |
Output is correct |
18 |
Correct |
3 ms |
600 KB |
Output is correct |
19 |
Correct |
2 ms |
440 KB |
Output is correct |
20 |
Correct |
2 ms |
344 KB |
Output is correct |
21 |
Correct |
2 ms |
344 KB |
Output is correct |
22 |
Correct |
1 ms |
344 KB |
Output is correct |
23 |
Correct |
8 ms |
444 KB |
Output is correct |
24 |
Correct |
5 ms |
344 KB |
Output is correct |
25 |
Correct |
11 ms |
440 KB |
Output is correct |
26 |
Correct |
16 ms |
708 KB |
Output is correct |
27 |
Correct |
38 ms |
600 KB |
Output is correct |
28 |
Correct |
5 ms |
600 KB |
Output is correct |
29 |
Correct |
41 ms |
444 KB |
Output is correct |
30 |
Correct |
24 ms |
604 KB |
Output is correct |
31 |
Correct |
10 ms |
500 KB |
Output is correct |
32 |
Correct |
11 ms |
700 KB |
Output is correct |
33 |
Correct |
14 ms |
692 KB |
Output is correct |
34 |
Correct |
11 ms |
600 KB |
Output is correct |
35 |
Correct |
11 ms |
856 KB |
Output is correct |
36 |
Correct |
13 ms |
700 KB |
Output is correct |
37 |
Correct |
15 ms |
596 KB |
Output is correct |
38 |
Correct |
16 ms |
1104 KB |
Output is correct |
39 |
Correct |
12 ms |
600 KB |
Output is correct |
40 |
Correct |
7 ms |
944 KB |
Output is correct |
41 |
Correct |
4 ms |
600 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
344 KB |
Output is correct |
4 |
Correct |
0 ms |
344 KB |
Output is correct |
5 |
Correct |
0 ms |
344 KB |
Output is correct |
6 |
Correct |
0 ms |
344 KB |
Output is correct |
7 |
Correct |
25 ms |
600 KB |
Output is correct |
8 |
Correct |
8 ms |
344 KB |
Output is correct |
9 |
Correct |
25 ms |
444 KB |
Output is correct |
10 |
Partially correct |
36 ms |
612 KB |
Output is partially correct |
11 |
Partially correct |
89 ms |
700 KB |
Output is partially correct |
12 |
Correct |
11 ms |
600 KB |
Output is correct |
13 |
Partially correct |
81 ms |
952 KB |
Output is partially correct |
14 |
Partially correct |
51 ms |
440 KB |
Output is partially correct |
15 |
Correct |
19 ms |
848 KB |
Output is correct |
16 |
Correct |
23 ms |
848 KB |
Output is correct |
17 |
Correct |
23 ms |
612 KB |
Output is correct |
18 |
Correct |
26 ms |
424 KB |
Output is correct |
19 |
Correct |
24 ms |
672 KB |
Output is correct |
20 |
Correct |
24 ms |
600 KB |
Output is correct |
21 |
Partially correct |
32 ms |
592 KB |
Output is partially correct |
22 |
Partially correct |
32 ms |
680 KB |
Output is partially correct |
23 |
Partially correct |
32 ms |
444 KB |
Output is partially correct |
24 |
Correct |
22 ms |
932 KB |
Output is correct |
25 |
Correct |
13 ms |
672 KB |
Output is correct |
26 |
Correct |
8 ms |
444 KB |
Output is correct |
27 |
Correct |
24 ms |
436 KB |
Output is correct |
28 |
Correct |
17 ms |
592 KB |
Output is correct |
29 |
Correct |
22 ms |
688 KB |
Output is correct |
30 |
Correct |
26 ms |
600 KB |
Output is correct |
31 |
Partially correct |
33 ms |
600 KB |
Output is partially correct |
32 |
Partially correct |
34 ms |
848 KB |
Output is partially correct |
33 |
Partially correct |
37 ms |
856 KB |
Output is partially correct |
34 |
Partially correct |
34 ms |
440 KB |
Output is partially correct |
35 |
Correct |
23 ms |
692 KB |
Output is correct |
36 |
Correct |
22 ms |
344 KB |
Output is correct |
37 |
Partially correct |
33 ms |
456 KB |
Output is partially correct |
38 |
Partially correct |
30 ms |
848 KB |
Output is partially correct |
39 |
Correct |
27 ms |
452 KB |
Output is correct |
40 |
Correct |
26 ms |
596 KB |
Output is correct |
41 |
Correct |
23 ms |
452 KB |
Output is correct |
42 |
Correct |
20 ms |
688 KB |
Output is correct |
43 |
Partially correct |
14 ms |
856 KB |
Output is partially correct |
44 |
Partially correct |
74 ms |
848 KB |
Output is partially correct |
45 |
Correct |
19 ms |
696 KB |
Output is correct |
46 |
Correct |
13 ms |
456 KB |
Output is correct |
47 |
Correct |
13 ms |
952 KB |
Output is correct |
48 |
Correct |
14 ms |
612 KB |
Output is correct |