Submission #256460

# Submission time Handle Problem Language Result Execution time Memory
256460 2020-08-02T17:48:26 Z A02 Game (IOI13_game) C++14
Compilation error
0 ms 0 KB
#include "robots.h"
#include <vector>
#include <algorithm>
#include <utility>
#include <set>
#include <iostream>

using namespace std;

int putaway(int A, int B, int T, int X[], int Y[], int W[], int S[]) {

    vector<int> weak_robots;
    vector<int> small_robots;

    for (int i = 0; i < A; i++){
        weak_robots.push_back(X[i]);
    }

    for (int i = 0; i < B; i++){
        small_robots.push_back(Y[i]);
    }

    sort(weak_robots.begin(), weak_robots.end());
    sort(small_robots.begin(), small_robots.end());

    for (int i = 0; i < T; i++){
        if (weak_robots.size() >= 1 && small_robots.size() >= 1){
            if (weak_robots[weak_robots.size() - 1] <= W[i] && small_robots[small_robots.size() - 1] <= S[i]){
                return -1;
            }
        }
        if (weak_robots.size() == 0){
            if (small_robots[small_robots.size() - 1] <= S[i]){
                return -1;
            }
        }
        if (small_robots.size() == 0){
            if (weak_robots[weak_robots.size() - 1] <= W[i]){
                return -1;
            }
        }
    }

    int put_away = 0;
    int time_taken = 0;

    multiset<pair<int, int> > toys_weight_first;
    multiset<pair<int, int> > toys_size_first;

    for (int i = 0; i < T; i++){
        pair<int, int> p1;
        pair<int, int> p2;

        p1.first = -W[i];
        p1.second = -S[i];

        p2.first = -S[i];
        p2.second = -W[i];

        toys_weight_first.insert(p1);
        toys_size_first.insert(p2);
    }

    while (put_away != T){

        time_taken++;

        for (int i = 0; i < A; i++){
            pair<int, int> current;
            current.first = -weak_robots[i];
            current.second = 0;

            multiset<pair<int, int> >::iterator it = toys_weight_first.lower_bound(current);

            if (it != toys_weight_first.end()){

                int w = it -> first;
                int s = it -> second;

                put_away++;

                toys_weight_first.erase(it);
                toys_size_first.erase(toys_size_first.lower_bound(make_pair(s, w)));

            }

        }

        for (int i = 0; i < B; i++){
            pair<int, int> current;
            current.first = -small_robots[i];
            current.second = 0;


            multiset<pair<int, int> >::iterator it = toys_size_first.lower_bound(current);

            if (it != toys_size_first.end()){

                int s = it -> first;
                int w = it -> second;

                put_away++;

                toys_size_first.erase(it);
                toys_weight_first.erase(toys_weight_first.lower_bound(make_pair(w, s)));
            }

        }

    }

    return time_taken;
}

Compilation message

grader.c: In function 'int main()':
grader.c:18:6: warning: variable 'res' set but not used [-Wunused-but-set-variable]
  int res;
      ^~~
game.cpp:1:10: fatal error: robots.h: No such file or directory
 #include "robots.h"
          ^~~~~~~~~~
compilation terminated.