제출 #578952

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
578952		Mazaalai	로봇 (IOI13_robots)	C++17	0 / 100	1 ms	468 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);
    }
};
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);
    set <Toy1> toys1, toys2;
    for (int i = 0; i < n; i++) {
        toys1.insert({w[i], sz[i], i});
        toys2.insert({sz[i], w[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);
                    toys2.erase(toys2.lb({sz[id], w[id], id}));
                    curRemoved = 1;
                    removeCnt++;
                    break;
                }
            }
        }
        if (small > 0) {
            Toy1 toy2 = {0, 0, 0};
            for (int i = 0; i < small; i++) {
                toy2.w = szLimit[i];
                while(!toys2.empty()) {
                    auto it = toys2.ub(toy2);
                    if (it == toys2.begin()) break;
                    it--;
                    int id = it->id;
                    toys2.erase(it);
                    toys1.erase(toys1.lb({sz[id], w[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: 0 / 14
• Subtask 2: 0 / 14
• Subtask 3: 0 / 25
• Subtask 4: 0 / 37
• Subtask 5: 0 / 10