#include <bits/stdc++.h>
#define pb push_back
typedef long long ll;
using namespace std;
const ll INF = (ll)(1e18);
struct segtree
{
vector<ll> smx, smn, sum;
int N;
segtree()
{
N = 1<<18;
smx.assign(2*N,0);
smn.assign(2*N,0);
sum.assign(2*N,0);
}
void upd(int i, int s, int e, int t, ll v)
{
if(s==e)
{
smx[i] = max(0LL, v);
smn[i] = min(0LL, v);
sum[i] = v;
return;
}
int m=(s+e)/2;
if(t<=m) upd(2*i, s, m, t, v);
else upd(2*i+1, m+1, e, t, v);
sum[i] = sum[2*i]+sum[2*i+1];
smx[i] = max(smx[2*i],sum[2*i]+smx[2*i+1]);
smn[i] = min(smn[2*i],sum[2*i]+smn[2*i+1]);
}
ll get(int i, int s, int e, ll c)
{
if(s==e) return min(max(sum[i], 0LL), c);
int m = (s+e)/2;
if(smx[2*i+1]-smn[2*i+1]>c) return get(2*i+1, m+1, e, c);
ll r = get(2*i,s,m,c);
if(r+smx[2*i+1]>c) return c-(smx[2*i+1]-sum[2*i+1]);
if(r+smn[2*i+1]<0) return sum[2*i+1]-smn[2*i+1];
return r+sum[2*i+1];
}
} st;
vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v)
{
int N=c.size(), Q=l.size();
vector<int> final_A;
vector<vector<int>> q(N+1);
for (int i=0; i<Q; i++) q[l[i]].pb(i), q[r[i]+1].pb(-i);
for (int i=0; i<N; i++)
{
for (auto j:q[i])
{
if (j>=0) st.upd(1, 0, Q-1, j, v[j]);
else st.upd(1, 0, Q-1, -j, 0);
}
final_A.pb(st.get(1, 0, Q-1, c[i]));
}
return final_A;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
12620 KB |
Output is correct |
2 |
Incorrect |
7 ms |
12624 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
453 ms |
30720 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
10 ms |
12620 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
8 ms |
12620 KB |
Output is correct |
2 |
Correct |
7 ms |
12604 KB |
Output is correct |
3 |
Correct |
130 ms |
19384 KB |
Output is correct |
4 |
Correct |
78 ms |
20252 KB |
Output is correct |
5 |
Correct |
203 ms |
26852 KB |
Output is correct |
6 |
Correct |
213 ms |
26948 KB |
Output is correct |
7 |
Correct |
211 ms |
26860 KB |
Output is correct |
8 |
Correct |
201 ms |
26928 KB |
Output is correct |
9 |
Correct |
218 ms |
26944 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
12620 KB |
Output is correct |
2 |
Incorrect |
7 ms |
12624 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |