#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){
PrevIndexIn.clear();
PrevIndexOut.clear();
//Pushing insects inside the machine, keeping track of the ones being putted inside for this iteration of binary searching
for(int i=0;i<RemainingInsects.size();i++){
move_inside(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=1;
PrevIndexIn.push_back(RemainingInsects[i]);
if(press_button()>cnt){
PrevIndexIn.pop_back();
PrevIndexOut.push_back(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=0;
move_outside(RemainingInsects[i]);
}
}
CurrentInMachine=0;
for(int i=0;i<n;i++){
CurrentInMachine+=InMachine[i];
}
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());
PrevIndexIn.pop_back();
}
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
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
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
PrevIndexIn.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=2;
//Phase 1: Determining the amount of different types of insects
TypeCount=0;
for(int i=0;i<n;i++){
move_inside(i);
InMachine[i]=1;
PrevIndexIn.push_back(i);
if(press_button()>1){
move_outside(i);
InMachine[i]=0;
PrevIndexIn.pop_back();
PrevIndexOut.push_back(i);
}else{
TypeCount++;
}
}
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
r=(n/TypeCount);
int cnt;
int ans=1;
//Phase 2: Binary search for the answer
while(r-l>1){
mid=(l+r)/2;
cnt=mid;
PrevIndexIn.clear();
PrevIndexOut.clear();
//Pushing insects inside the machine, keeping track of the ones being putted inside for this iteration of binary searching
for(int i=0;i<RemainingInsects.size();i++){
move_inside(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=1;
PrevIndexIn.push_back(RemainingInsects[i]);
if(press_button()>cnt){
PrevIndexIn.pop_back();
PrevIndexOut.push_back(RemainingInsects[i]);
InMachine[RemainingInsects[i]]=0;
move_outside(RemainingInsects[i]);
}
}
CurrentInMachine=0;
for(int i=0;i<n;i++){
CurrentInMachine+=InMachine[i];
}
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());
PrevIndexIn.pop_back();
}
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
mid=r-1;
}else if(CurrentInMachine==cnt*TypeCount){
//As every type of insects has at least cnt insects, the answer is at least cnt. Just return true
for(int i=0;i<n;i++){
if(!InMachine[i]){
RemainingInsects.push_back(i);
}
}
PrevIndexIn.clear();
mid=l;
ans=mid;
}
}
if(PossibleMincount(r)){
return r;
}else{
return l;
}
}
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: In function 'int min_cardinality(int)':
insects.cpp:97:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
97 | for(int i=0;i<RemainingInsects.size();i++){
| ~^~~~~~~~~~~~~~~~~~~~~~~~
insects.cpp:89:6: warning: variable 'ans' set but not used [-Wunused-but-set-variable]
89 | int ans=1;
| ^~~
insects.cpp: In function 'int PossibleMincount(int)':
insects.cpp:55:1: warning: control reaches end of non-void function [-Wreturn-type]
55 | }
| ^
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
344 KB |
Output is correct |
| 2 |
Correct |
0 ms |
384 KB |
Output is correct |
| 3 |
Correct |
0 ms |
344 KB |
Output is correct |
| 4 |
Correct |
1 ms |
344 KB |
Output is correct |
| 5 |
Incorrect |
0 ms |
344 KB |
Wrong answer. |
| 6 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
344 KB |
Output is correct |
| 2 |
Correct |
0 ms |
384 KB |
Output is correct |
| 3 |
Correct |
0 ms |
344 KB |
Output is correct |
| 4 |
Correct |
1 ms |
344 KB |
Output is correct |
| 5 |
Incorrect |
0 ms |
344 KB |
Wrong answer. |
| 6 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
340 KB |
Wrong answer. |
| 2 |
Halted |
0 ms |
0 KB |
- |
