Submission #123762

#	Time	Username	Problem	Language	Result	Execution time	Memory
123762		MAMBA	Robots (IOI13_robots)	C++17	76 / 100	3035 ms	8424 KiB

robots

#include <bits/stdc++.h>
#include "robots.h"

using namespace std;

#define rep(i , j , k) for (int i = j; i < (int)k; i++)

const int MAXN = 1e6 + 10;

int a[MAXN], w[MAXN];
priority_queue<int> st;

int putaway(int A, int B, int T, int X[], int Y[], int W[], int S[]) {
   
	iota(a , a + T , 0);

	rep(i , 0 , T) w[i]= W[i];

	sort(a , a + T, [](int l , int r) { return w[l] < w[r]; });

	sort(X , X + A);
	sort(Y , Y + B, greater<int>());

	int lo = -1, hi = T + 1;
	while (lo != hi - 1) {
		int mid = lo + hi >> 1;

		while (st.size()) st.pop();
		int ptr = 0;
		rep(i , 0 , A) {
			while (ptr < T && W[a[ptr]] < X[i]) 
				st.push(S[a[ptr++]]);
			rep(j , 0 , mid)
				if (st.size()) 
					st.pop();
		}
		while (ptr < T)
			st.push(S[a[ptr++]]);
		rep(i , 0 ,B)
			rep(j , 0 , mid)
				if (st.size() && st.top() < Y[i])
					st.pop();
		if (st.empty())
			hi = mid;
		else 
			lo = mid;
	}
	if (hi == T + 1) hi = -1;
	return hi;
}

Compilation message (stderr)

robots.cpp: In function 'int putaway(int, int, int, int*, int*, int*, int*)':
robots.cpp:26:16: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   int mid = lo + hi >> 1;
             ~~~^~~~

• Subtask 1: 14 / 14
• Subtask 2: 0 / 14
• Subtask 3: 25 / 25
• Subtask 4: 37 / 37
• Subtask 5: 0 / 10