Submission #978133

#	Time	Username	Problem	Language	Result	Execution time	Memory
978133		ThegeekKnight16	Robots (IOI13_robots)	C++17	100 / 100	1330 ms	26652 KiB

robots

#include <bits/stdc++.h>
#include "robots.h"
using namespace std;
#pragma GCC optimize("O3")
#pragma GCC target("avx2")

bool Test(int x, int A, int B, int T, int *X, int *Y, const vector<pair<int, int>> &Toy)
{
  if ((long long)x*(A+B) < T) return 0;
  int idT = 0, idA = 0;
  priority_queue<int> remain;
  while (idT < T && idA < A)
  {
    if (X[idA] <= Toy[idT].first)
    {
      int remov = x;
      while (remov > 0 && !remain.empty()) remain.pop(), remov--;
      idA++;
    }
    else remain.push(Toy[idT++].second);
  }
  while (idT < T) remain.push(Toy[idT++].second);
  while (idA < A)
  {
    int remov = x;
    while (remov > 0 && !remain.empty()) remain.pop(), remov--;
    idA++;
  }

  vector<int> bonga;
  while (!remain.empty()) bonga.push_back(remain.top()), remain.pop();
  reverse(bonga.begin(), bonga.end());
  auto idR = 0; int idB = 0;
  int cnt = 0;
  while (idR < bonga.size() && idB < B)
  {
    if (Y[idB] <= bonga[idR]) cnt = max(0, cnt-x), idB++;
    else cnt++, idR++;
  }
  if (idR < bonga.size()) return 0;
  while (idB < B) cnt = max(0, cnt-x), idB++;

  return (cnt == 0);
}

int putaway(int A, int B, int T, int X[], int Y[], int _W[], int _S[])
{
  vector<pair<int, int>> Toy(T);
  for (int i = 0; i < T; i++) Toy[i] = make_pair(_W[i], _S[i]);
  sort(X, X+A); sort(Y, Y+B);
  for (int i = 0; i < T; i++) if (Toy[i].first >= X[A-1] && Toy[i].second >= Y[B-1]) return -1;
  sort(Toy.begin(), Toy.end());

  int ini = 1, fim = T;
  while (ini < fim)
  {
    int m = (ini + fim) >> 1;
    if (Test(m, A, B, T, X, Y, Toy)) fim = m;
    else ini = m+1;
  }
  return fim;
}

Compilation message (stderr)

robots.cpp: In function 'bool Test(int, int, int, int, int*, int*, const std::vector<std::pair<int, int> >&)':
robots.cpp:35:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |   while (idR < bonga.size() && idB < B)
      |          ~~~~^~~~~~~~~~~~~~
robots.cpp:40:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |   if (idR < bonga.size()) return 0;
      |       ~~~~^~~~~~~~~~~~~~

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