Submission #966187

#	Time	Username	Problem	Language	Result	Execution time	Memory
966187		AkibAzmain	Distributing Candies (IOI21_candies)	C++17	29 / 100	150 ms	32572 KiB

candies

#include "candies.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;

std::vector<int> distribute_candies (std::vector<int> c, std::vector<int> l,
                                     std::vector<int> r, std::vector<int> v)
{
  ll n = c.size();
  v.push_back (0);
  vector < ll > a;
  for (auto &x : v) a.push_back (x);
  reverse (a.begin (), a.end ());
  a.push_back (-1e18);
  for (auto &x : a) x *= -1;
  for (int i = 1; i < a.size (); ++i) a[i] += a[i - 1];
  pair < ll, ll > mn = { 0, 0 }, mx = { 0, 0 };
  vector < pair < ll, ll > > mna (a.size ()), mxa = mna;
  map < ll, int > mp;
  for (int i = 0; i < a.size (); ++i)
    {
      if (a[i] < mn.first) mn = { a[i], i };
      if (a[i] > mx.first) mx = { a[i], i };
      mna[i] = mn;
      mxa[i] = mx;
      if (mp.count (mx.first - mn.first) == 0)
        mp[mx.first - mn.first] = i;
    }
  vector < int > ans (n);
  for (int i = 0; i < n; ++i)
    {
      int j = mp.lower_bound (c[i])->second;
      if (a[j] == mna[j].first)
        ans[i] = c[i] - a[mxa[j].second];
      else
        ans[i] = -a[mna[j].second];
      if (ans[i] < 0) ans[i] += c[i];
    }
  return ans;
}

Compilation message (stderr)

candies.cpp: In function 'std::vector<int> distribute_candies(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
candies.cpp:16:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 |   for (int i = 1; i < a.size (); ++i) a[i] += a[i - 1];
      |                   ~~^~~~~~~~~~~
candies.cpp:20:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |   for (int i = 0; i < a.size (); ++i)
      |                   ~~^~~~~~~~~~~

• Subtask 1: 0 / 3
• Subtask 2: 0 / 8
• Subtask 3: 0 / 27
• Subtask 4: 29 / 29
• Subtask 5: 0 / 33