#include "candies.h"
#include <algorithm>
#include <bitset>
#include <cstdint>
#include <cstring>
#include <iostream>
#include <limits.h>
#include <math.h>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
#define ll long long
#define loop(X, N) for(int X = 0; X < (N); X++)
#define all(V) V.begin(), V.end()
#define rall(V) V.rbegin(), V.rend()
using namespace std;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef vector<vector<ii>> vvii;
typedef vector<bool> vb;
typedef vector<vector<bool>> vvb;
// struct SegTree {
// int n, N;
// vector<set<int>> tree;
// SegTree(int _n) {
// int n = _n;
// int N = 1;
// while (N < n) N *= 2;
// tree.resize(2 * N);
// }
// };
vi distribute_candies(vi c, vi ls, vi rs, vi vs) {
int n = c.size();
int q = ls.size();
vector<ll> cumul(q, vs[0]);
vector<ll> sufMax(q), sufMin(q);
for (int i = 1; i < q; i++) {
ll v = vs[i];
cumul[i] = cumul[i - 1] + v;
}
sufMax.back() = cumul.back();
sufMin.back() = cumul.back();
for (int i = q - 2; i >= 0; i--) {
sufMin[i] = min(sufMin[i + 1], cumul[i]);
sufMax[i] = max(sufMax[i + 1], cumul[i]);
}
vi s(n);
loop(i, n) {
int l = 0, r = q - 1;
while (l < r) {
int m = (l + r) / 2;
ll diff = sufMax[m] - sufMin[m];
if (diff <= c[i]) {
r = m;
}
else {
l = m + 1;
}
}
ll intermediate = cumul[l];
intermediate = min(intermediate, (ll)c[i]);
intermediate = max(intermediate, (ll)0);
s[i] = intermediate + cumul.back() - cumul[l];
s[i] = min(s[i], c[i]);
s[i] = max(s[i], 0);
}
return s;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
78 ms |
12112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |