#include<bits/stdc++.h>
#define fi first
#define se second
#define endl "\n"
#define ii pair<int, int>
using namespace std;
const int N = 1e5 + 10;
vector<int> event[N];
vector<long long> distribute_candies (vector<int> c, vector<int> l, vector<int> r, vector<int> v) {
vector<long long> p;
int n = c.size();
p.resize(n);
for(int i = 0; i < n; i++) event[i].push_back(0);
for(int i = 0; i < v.size(); i++)
{
for(int j = l[i]; j <= r[i]; j++)
{
event[j].push_back(v[i]);
}
}
for(int i = 0; i < n; i++)
{
long long sum = 0, border = 0;
vector<long long> pref;
for(int &j : event[i]) {
sum += j;
pref.push_back(sum);
}
long long mx = -1e18, mn = 1e18;
for(int j = pref.size() - 1; j >= 0; j--)
{
mn = min(mn, pref[j]);
mx = max(mx, pref[j]);
if (mx - mn <= c[i])
{
if (event[i][j] > 0 && mx <= pref[j]) border = pref[j] - c[i];
if (event[i][j] <= 0 && mn >= pref[j]) border = pref[j];
}
}
p.push_back(sum - border);
// cout << sum - border << " ";
}
return p;
}
#ifdef ngu
int main() {
freopen ("task.inp", "r", stdin);
freopen ("task.out", "w", stdout);
int n, q; cin >> n >> q;
vector<int> c(n), l(q), r(q), v(q);
for(int i = 0; i < n; i++) cin >> c[i];
for(int i = 0; i < q; i++) cin >> l[i] >> r[i] >> v[i];
vector<long long> ans = distribute_candies(c, l, r, v);
// for(int &j : ans) cout << j << " ";
}
#endif // ngu
Compilation message
candies.cpp: In function 'std::vector<long long int> distribute_candies(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
candies.cpp:20:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
20 | for(int i = 0; i < v.size(); i++)
| ~~^~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
74 ms |
27416 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
2768 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |