#include <bits/stdc++.h>
#include "insects.h"
#define pb push_back
using namespace std;
#include "../debug.h"
//#define debug(x...) void(42)
constexpr static int BLOCK = 100;
int n;
set<int> basis;
bool check(int val)
{
vector<int> ci;
for (int i = 0; i < n; i++)
{
if (basis.find(i) != basis.end())
continue;
move_inside(i);
ci.pb(i);
if (press_button() > val)
{
ci.pop_back();
move_outside(i);
}
}
bool res = ci.size() == (val - 1) * basis.size();
for (int i : ci)
move_outside(i);
return res;
}
int test_a()
{
int l = 1, r = 1 + (n / basis.size());
debug(l, r);
while (r - l > 1)
{
int m = l+r>>1;
debug(m);
if(check(m))
l = m;
else
r = m;
}
debug(l);
return l;
}
vector<int> bs;
set<int> machine;
int get_min(int l, int r, vector<int>& v)
{
if (r - l == 1)
{
debug(l, v);
return v.size()+1;
}
int m = l+r>>1;
for (int i = l; i < m; i++)
{
if (machine.find(bs[i]) != machine.end())
continue;
machine.insert(bs[i]);
move_inside(bs[i]);
}
for (int i = m; i < r; i++)
{
if (machine.find(bs[i]) == machine.end())
continue;
machine.erase(bs[i]);
move_outside(bs[i]);
}
vector<int> lv, rv;
for (int i : v)
{
move_inside(i);
if (press_button() > 1)
lv.pb(i);
else
rv.pb(i);
move_outside(i);
}
int lres = get_min(l, m, lv);
int rres = get_min(m, r, rv);
return min(lres, rres);
}
int test_b()
{
for (int i : basis)
{
bs.pb(i);
machine.insert(i);
}
vector<int> v;
for (int i = 0; i < n; i++)
if (basis.find(i) == basis.end())
v.pb(i);
return get_min(0, bs.size(), v);
}
int min_cardinality(int N)
{
n = N;
for (int i = 0; i < n; i++)
{
move_inside(i);
if (press_button() > 1)
move_outside(i);
else
basis.insert(i);
}
if (basis.size() > BLOCK)
return test_a();
else
return test_b();
}
Compilation message
insects.cpp:6:10: fatal error: ../debug.h: No such file or directory
6 | #include "../debug.h"
| ^~~~~~~~~~~~
compilation terminated.
