Submission #587251

#	Time	Username	Problem	Language	Result	Execution time	Memory
587251		jakubd	Distributing Candies (IOI21_candies)	C++17	3 / 100	109 ms	8248 KiB

candies

#include "candies.h"
 
#include <bits/stdc++.h>
 
using namespace std;
 
vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v) {
  int n = c.size(), q = (int)l.size();
  vector<int> s;
 
  if (n <= 2000 && q <= 2000) {
    s.resize(n);
 
    for (int i = 0; i < q; i++) {
      for (int j = l[i]; j <= r[i]; j++) {
        s[j] += v[i];
 
        if (s[j] < 0) s[j] = 0;
        if (s[j] > c[j]) s[j] = c[j];
      }
    }
  } else {
    s.resize(n + 1);
 
    for (int i = 0; i < q; i++) {
      s[l[i]] += v[i];
      s[r[i] + 1] -= v[i];
    }
 
    for (int i = 0; i < n; i++) s[i + 1] += s[i];
 
    vector<int> res(n);
    for (int i = 0; i < n; i++) {
      res[i] = min(s[i + 1], c[i]);
    }

    s = res;
  }
 
  return s;
}

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