47 lines
777 B
C
47 lines
777 B
C
struct SegmentTree {
|
|
int N;
|
|
int * seg;
|
|
}
|
|
|
|
// Build segment tree
|
|
void Build(int l, int r, int n) {
|
|
if (l == r) {
|
|
|
|
}
|
|
int m = l + r >> 1;
|
|
Build(l, m, n << 1);
|
|
Build(m + 1, r, n << 1 | 1);
|
|
seg[n] = seg[n << 1] + seg[n << 1 | 1];
|
|
}
|
|
|
|
// Update value at index x
|
|
void Update(int x, int v, int l, int r, int n) {
|
|
if (l == r) {
|
|
seg[n] += v;
|
|
return;
|
|
}
|
|
int m = l + r >> 1;
|
|
if (x <= m) {
|
|
Update(x, v, l, m, n<<1);
|
|
}
|
|
else {
|
|
Update(x, v, m+1, r, n<<1|1);
|
|
}
|
|
seg[n] = seg[n << 1] + seg[n << 1 | 1];
|
|
}
|
|
|
|
// Find element with prefix sum v
|
|
int Query(int v, int l, int r, int n) {
|
|
if (l == r) {
|
|
return s.seg[n], l;
|
|
}
|
|
int m = l + r >> 1;
|
|
if (seg[n << 1] >= v) {
|
|
return Query(v, l, m, n << 1);
|
|
}
|
|
else {
|
|
return Query(v - s.seg[n << 1], m + 1, r, n << 1 | 1);
|
|
}
|
|
}
|
|
|