Submission #830694

#	Time	Username	Problem	Language	Result	Execution time	Memory
830694		caganyanmaz	드문 곤충 (IOI22_insects)	C++17	0 / 100	99 ms	336 KiB

insects

#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 = 1;

int n;
set<int> basis;

bool check(int val)
{
	vector<int> ci;
	int counter = basis.size();
	for (int i = 0; i < n; i++)
	{
		if (basis.find(i) != basis.end())
			continue;
		move_inside(i);
		counter++;
		if (press_button() > val)
		{
			move_outside(i);
			counter--;
		}
	}
	return counter >= val * basis.size();
}

int test_a()
{
	int l = -1, r = 5 + (n / basis.size());
	while (r - l > 1)
	{
		int m = l+r>>1;
		if(check(m))
			l = m;
		else
			r = m;
	}
	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 = 0; i < n; i++)
	{
		if (i >= l && i < m)
		{
			if (machine.find(bs[i]) != machine.end())
				continue;
			machine.insert(bs[i]);
			move_inside(bs[i]);
		}
		if (i < l || i >= m)
		{
			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 (stderr)

insects.cpp: In function 'bool check(int)':
insects.cpp:30:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::set<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |  return counter >= val * basis.size();
      |         ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
insects.cpp: In function 'int test_a()':
insects.cpp:38:12: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   38 |   int m = l+r>>1;
      |           ~^~
insects.cpp: In function 'int get_min(int, int, std::vector<int>&)':
insects.cpp:56:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   56 |  int m = l+r>>1;
      |          ~^~

• Subtask 1: 0 / 10
• Subtask 2: 0 / 15
• Subtask 3: 0 / 75