#include "candies.h"
#include "bits/stdc++.h"
using namespace std;
#define forR(i, x) for(int i = 0; i < (x); ++i)
#define REP(i, a, b) for(int i = (a); i < (b); ++i)
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;
std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
std::vector<int> r, std::vector<int> v) {
int n=c.size(), m=l.size();
vi ch(m);
vvi addOp(n), remOp(n);
forR(i, m) {
addOp[l[i]].push_back(i);
if(r[i] + 1 < n) remOp[r[i]+1].push_back(i);
}
vi ret(n);
vi tested(n);
forR(i, n) {
for(int j : addOp[i]) ch[j] = v[j];
for(int j : remOp[i]) ch[j] = 0;
// [besInd, n) has NO interval <= -c or >= c
int besInd=n;
ll uc=0, dc=0;
for(int j = n-1; j >= 0; --j) {
uc = max(uc + ch[j], 0ll);
dc = min(dc + ch[j], 0ll);
if(uc >= c[i] || dc <= -c[i]) break;
else besInd = j;
}
// cerr << i << ": \n";
// for(int j : ch) cerr << j << ' ';
// cerr << '\n';
// cerr << besInd << ' ' << uc << ' ' << dc << '\n';
if(uc >= c[i]) {
assert(dc >= -c[i]);
int cur = c[i];
REP(j, besInd, n) {
cur = min(c[i], cur + ch[j]);
assert(cur >= 0);
}
ret[i] = cur;
} else {
assert(besInd == 0 || dc <= -c[i]);
int cur = 0;
REP(j, besInd, n) {
cur = max(0, cur + ch[j]);
assert(cur <= c[i]);
}
ret[i] = cur;
}
int cur = 0;
forR(j, m) cur = max(0, min(c[i], cur + ch[j]));
assert(cur == ret[i]);
}
return ret;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Runtime error |
1 ms |
604 KB |
Execution killed with signal 6 |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
5052 ms |
25444 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1 ms |
604 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Runtime error |
1 ms |
604 KB |
Execution killed with signal 6 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Runtime error |
1 ms |
604 KB |
Execution killed with signal 6 |
4 |
Halted |
0 ms |
0 KB |
- |