#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();
int q = l.size();
vector<int> s(n, 0);
if (n <= 3000 && q <= 3000) {
for (int i = 0; i < q; ++i) {
int L = l[i];
int R = r[i];
int V = v[i];
for (int j = L; j <= R; ++j) {
if (V > 0) s[j] = min(c[j], s[j] + V);
else s[j] = max(0, s[j] + V);
}
}
} else {
vector<pair<int, int>> events;
for (int i = 0; i < q; ++i) {
int L = l[i];
int R = r[i];
int V = v[i];
events.push_back({L, V});
events.push_back({R + 1, -V});
}
sort(events.begin(), events.end());
int ptr = 0;
int sum = 0;
for (int i = 0; i < n; ++i) {
while (ptr < 2 * q && events[ptr].first <= i) sum += events[ptr++].second;
s[i] = min(c[i], sum);
}
}
return s;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |