Submission #579006

#	Time	Username	Problem	Language	Result	Execution time	Memory
579006		Mazaalai	Robots (IOI13_robots)	C++17	100 / 100	1734 ms	14504 KiB

robots

#include "robots.h"
#include <bits/stdc++.h>
#define ub upper_bound
#define lb lower_bound
#define mp make_pair
#define ff first
#define ss second
#define LINE "--------------------\n"
using namespace std;
using PII = pair <int, int>;
int n, m, k, a, b;
const int N = 5e4+5;
const int M = 1e6+5;
int wLimit[N], szLimit[N];
PII toys[M];
bool solve(int lim) {
    priority_queue <int> vals;
    int i, j;
    for (i = 0, j = 0; i < a; i++) {
        while(j < n && wLimit[i] > toys[j].ff) vals.push(toys[j++].ss);
        for (int p = 0; p < lim && !vals.empty(); p++) vals.pop();
    }    
    while(j < n) vals.push(toys[j++].ss);
    for (int i = b-1; i >= 0; i--) {
        if (!vals.empty() && !(vals.top() < szLimit[i])) return 0;
        for (int j = 0; j < lim && !vals.empty(); j++) vals.pop();
    }
    return vals.empty();
}
int putaway(int weak, int small, int _n, int _wLimit[], int _szLimit[], int _w[], int _sz[]) {
    n = _n;
    a = weak;
    b = small;
    for (int i = 0; i < weak; i++) wLimit[i] = _wLimit[i];
    for (int i = 0; i < small; i++) szLimit[i] = _szLimit[i];
    for (int i = 0; i < n; i++) toys[i] = {_w[i], _sz[i]};
    if (n > 0) sort(toys, toys+n);
    if (a > 0) sort(wLimit, wLimit+a); 
    if (b > 0) sort(szLimit, szLimit+b);
    if (n == 0) return 0;
    int l = 1, r = n, ans = 1e9;
    // l = r = 3;
    while (l <= r) {
        int mid = (l+r)>>1;
        if (solve(mid)) {
            // cout << "A\n";
            r = mid-1;
            ans = mid;
        } else {
            // cout << "B\n";
            l = mid+1;
        }
    }
    return ans == 1e9 ? -1 : ans;
}

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