Submission #781913

#TimeUsernameProblemLanguageResultExecution timeMemory
781913caganyanmazRobots (IOI13_robots)C++17
76 / 100
3093 ms54276 KiB
#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include "robots.h"
#define DEBUGGING
using namespace std;


constexpr static int DEBUG_PRINT_SIZE = 10;
void __print(int i) { cerr << i; }
void __print(const char* c) { cerr << c; }
void __print(int* i) { cerr << "{"; for (int j = 0; j < DEBUG_PRINT_SIZE; j++) { cerr << (j ? ", " : ""); __print(i[j]); } cerr << "}"; }
template<typename T>
void __print(T& t) { cerr << "{"; int f = 0; for (auto& i : t) { cerr << (f++ ? ", " : ""); __print(i); } cerr << "}"; }
void _print() { cerr << "]\n"; }
template<typename T, typename... V>
void _print(T t, V... v) { __print(t); if (sizeof...(v)) cerr << ", "; _print(v...); }
#ifdef DEBUGGING
#define debug(x...) cerr << "[" << (#x) << "] = ["; _print(x)
#else
#define debug(x...)
#endif

bool is_possible(int a, int b, int t, int x[], int y[], int k);
int ceil_div(int a, int b);


constexpr static int MXTOYSIZE = 1e6;
constexpr static int MXSIZE = 50000;
multiset<int> xdata[MXSIZE+1];
int putaway(int a, int b, int t, int x[], int y[], int ww[], int ss[])
{
	sort(x, x + a);
	sort(y, y + b);
	for (int i = 0; i < t; i++)
	{
		int index = upper_bound(x, x + a, ww[i]) - x;
		xdata[index].insert(ss[i]);
	}
	int l = 0, r = t+1;
	while (r - l > 1)
	{
		int m = l+r>>1;
		if (is_possible(a, b, t, x, y, m))
			r = m;
		else
			l = m;
	}
	if (r == t+1)
		return -1;
	return r;
}

int ycounts[MXSIZE];
bool insert_to_ycounts(int y[], int b, int val);
bool is_possible(int a, int b, int t, int x[], int y[], int k)
{
	for (int i = 0; i < b; i++)
		ycounts[i] = 0;
	multiset<int> m;
	for (int i = a; i >= 0; i--)
	{
		m.insert(xdata[i].begin(), xdata[i].end());
		while (m.size() && (i == a || ceil_div(m.size(), a - i) >  k))
			if (!insert_to_ycounts(y, b, *m.begin()))
				return false;
			else
				m.erase(m.begin());
	}
	int j = t-1;
	int prefix = 0;
	for (int i = b-1; i >= 0; i--)
	{
		prefix += ycounts[i];
		if (ceil_div(prefix, b - i) > k)
			return false;
	}
	return true;
}
bool insert_to_ycounts(int y[], int b, int val)
{
	int index = upper_bound(y, y + b, val) - y;
	if (index == b)
		return false;
	ycounts[index]++;
	return true;
}


int ceil_div(int a, int b)
{
	return (a + b - 1) / b;
}

Compilation message (stderr)

robots.cpp: In function 'int putaway(int, int, int, int*, int*, int*, int*)':
robots.cpp:43:12: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   43 |   int m = l+r>>1;
      |           ~^~
robots.cpp: In function 'bool is_possible(int, int, int, int*, int*, int)':
robots.cpp:70:6: warning: unused variable 'j' [-Wunused-variable]
   70 |  int j = t-1;
      |      ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...