Submission #578942

#	Time	Username	Problem	Language	Result	Execution time	Memory
578942		Mazaalai	Robots (IOI13_robots)	C++17	39 / 100	910 ms	65536 KiB

robots

#include "robots.h"
#include <bits/stdc++.h>
#define ub upper_bound
#define lb lower_bound
#define mp make_pair
#define LINE "--------------------\n"
using namespace std;
int n, m, k;
struct Toy1 {
    int w, sz, id;
    bool operator < (const Toy1& a) const {
        return mp(w, sz) < mp(a.w, a.sz);
    }
};
struct Toy2 {
    int w, sz, id;
    bool operator < (const Toy2& a) const {
        return mp(sz, w) < mp(a.sz, a.w);
    }
};
int putaway(int weak, int small, int n, int wLimit[], int szLimit[], int w[], int sz[]) {
    sort(wLimit, wLimit+weak);
    sort(szLimit, szLimit+small);
    multiset <Toy1> toys1;
    multiset <Toy2> toys2;
    for (int i = 0; i < n; i++) {
        toys1.insert({w[i], sz[i], i});
        toys2.insert({w[i], sz[i], i});
    }
    int ans = 0, removeCnt = 0;
    while(removeCnt < n) {
        // cout << LINE;
        // for (auto el : toys1) {
        //     cout << el.w << ' ' << el.sz << ' ' << el.id << '\n';
        // }
        // cout << "\n";
        // for (auto el : toys2) {
        //     cout << el.w << ' ' << el.sz << ' ' << el.id << '\n';
        // }
        ans++;
        bool curRemoved = 0;
        Toy1 toy1 = {0, 0, 0};
        for (int i = 0; i < weak; i++) {
            toy1.w = wLimit[i];
            while(!toys1.empty()) {
                auto it = toys1.ub(toy1);
                if (it == toys1.begin()) break;
                it--;
                int id = it->id;
                toys1.erase(it);
                // cout << wLimit[i] << " remove : " << w[i] << ',' << sz[i] << '\n';
                toys2.erase(toys2.lb({w[id], sz[id], id}));
                curRemoved = 1;
                removeCnt++;
                break;
            }
        }
        Toy2 toy2 = {0, 0, 0};
        for (int i = 0; i < small; i++) {
            toy2.sz = szLimit[i];
            while(!toys2.empty()) {
                auto it = toys2.ub(toy2);
                if (it == toys2.begin()) break;
                it--;
                int id = it->id;
                toys2.erase(it);
                // cout << szLimit[i] << " remove : " << w[i] << ',' << sz[i] << '\n';
                toys1.erase(toys1.lb({w[id], sz[id], id}));
                curRemoved = 1;
                removeCnt++;
                break;
            }
        }
        if (!curRemoved) return -1;
    }
        // cout << LINE;
        // for (auto el : toys1) {
        //     cout << el.w << ' ' << el.sz << ' ' << el.id << '\n';
        // }
        // cout << "\n";
        // for (auto el : toys2) {
        //     cout << el.w << ' ' << el.sz << ' ' << el.id << '\n';
        // }
    return ans;
}

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