#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+1), q[r[i]+1].pb(-(i+1));
for (int i=0; i<N; i++)
{
for (auto j:q[i])
{
if (j>0) st.upd(1, 0, Q-1, j-1, v[j]);
else st.upd(1, 0, Q-1, -j-1, 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 |
12620 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
472 ms |
35168 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
7 ms |
12672 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
6 ms |
12620 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
12620 KB |
Output is correct |
2 |
Incorrect |
7 ms |
12620 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |