Submission #578948

#	Time	Username	Problem	Language	Result	Execution time	Memory
578948		Mazaalai	Robots (IOI13_robots)	C++14	39 / 100	815 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[]) {
    if (weak > 0) sort(wLimit, wLimit+weak);
    if (small > 0) 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;
        if (weak > 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;
                }
            }
        }
        if (small > 0) {
            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