Mountain
Problem Statement Summary Maintain a function f on discrete points \ 1,, N\, initially f(x) = 0. Support three operations: Range add: given l, r, v, set f(x) f(x) + v for x [l, r]. Clamp to zero: set f(x) (f(x), 0) fo...
Problem Statement
No standalone statement file is available for this entry.
A separate statement file is not available for this entry, so the page focuses on the editorial and implementation.
Editorial
The solution write-up is rendered from the LaTeX source, with equations kept live through MathJax.
Problem Statement Summary
Maintain a function $f$ on discrete points $\{1, \ldots, N\}$, initially $f(x) = 0$. Support three operations:
Range add: given $l, r, v$, set $f(x) \gets f(x) + v$ for $x \in [l, r]$.
Clamp to zero: set $f(x) \gets \max(f(x), 0)$ for all $x$.
Threshold query: given $h$, find the leftmost $x$ with $f(x) \ge h$, or report that none exists.
Solution: Segment Tree with Lazy Propagation
Lazy Tags
Each node carries two lazy tags:
lazy_add: pending additive update.lazy_set/set_val: pending ``set all to value'' (from the clamp operation).When pushing down, the ``set'' tag is applied first (overriding children), then the ``add'' tag.
Clamp Operation
The ``clamp to zero'' is handled recursively:
If $\min \ge 0$ in the segment: no change.
If $\max \le 0$: set the entire segment to 0.
Otherwise: push down lazy tags and recurse into children.
This is a form of the Segment Tree Beats technique. Segments that are entirely non-negative or entirely non-positive are handled in $O(1)$; only mixed segments require recursion.
Threshold Query
To find the leftmost $x$ with $f(x) \ge h$: if the segment's max is $< h$, return $-1$. Otherwise, recurse left first; if not found, recurse right.
C++ Implementation
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000005;
struct Node {
long long mn, mx;
long long lazy_add;
bool lazy_set;
long long set_val;
};
Node tree[4 * MAXN];
int n;
void build(int v, int l, int r) {
tree[v] = {0, 0, 0, false, 0};
if (l == r) return;
int mid = (l + r) / 2;
build(v * 2, l, mid);
build(v * 2 + 1, mid + 1, r);
}
void pushDown(int v, int l, int r) {
if (l == r) return;
for (int c : {v * 2, v * 2 + 1}) {
if (tree[v].lazy_set) {
tree[c].mn = tree[c].mx = tree[v].set_val;
tree[c].lazy_add = 0;
tree[c].lazy_set = true;
tree[c].set_val = tree[v].set_val;
}
tree[c].mn += tree[v].lazy_add;
tree[c].mx += tree[v].lazy_add;
if (tree[c].lazy_set)
tree[c].set_val += tree[v].lazy_add;
else
tree[c].lazy_add += tree[v].lazy_add;
}
tree[v].lazy_add = 0;
tree[v].lazy_set = false;
}
void pull(int v) {
tree[v].mn = min(tree[v * 2].mn, tree[v * 2 + 1].mn);
tree[v].mx = max(tree[v * 2].mx, tree[v * 2 + 1].mx);
}
void rangeAdd(int v, int l, int r, int ql, int qr, long long val) {
if (ql > r || qr < l) return;
if (ql <= l && r <= qr) {
tree[v].mn += val;
tree[v].mx += val;
if (tree[v].lazy_set) tree[v].set_val += val;
else tree[v].lazy_add += val;
return;
}
pushDown(v, l, r);
int mid = (l + r) / 2;
rangeAdd(v * 2, l, mid, ql, qr, val);
rangeAdd(v * 2 + 1, mid + 1, r, ql, qr, val);
pull(v);
}
void clampZero(int v, int l, int r) {
if (tree[v].mn >= 0) return;
if (tree[v].mx <= 0) {
tree[v].mn = tree[v].mx = 0;
tree[v].lazy_add = 0;
tree[v].lazy_set = true;
tree[v].set_val = 0;
return;
}
if (l == r) {
tree[v].mn = tree[v].mx = max(tree[v].mn, 0LL);
return;
}
pushDown(v, l, r);
int mid = (l + r) / 2;
clampZero(v * 2, l, mid);
clampZero(v * 2 + 1, mid + 1, r);
pull(v);
}
int query(int v, int l, int r, int ql, int qr, long long h) {
if (ql > r || qr < l || tree[v].mx < h) return -1;
if (l == r) return l;
pushDown(v, l, r);
int mid = (l + r) / 2;
int res = query(v * 2, l, mid, ql, qr, h);
if (res != -1) return res;
return query(v * 2 + 1, mid + 1, r, ql, qr, h);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int Q;
cin >> n >> Q;
build(1, 1, n);
while (Q--) {
char op;
cin >> op;
if (op == 'A') {
int l, r;
long long val;
cin >> l >> r >> val;
rangeAdd(1, 1, n, l, r, val);
} else if (op == 'M') {
clampZero(1, 1, n);
} else if (op == 'Q') {
long long h;
cin >> h;
cout << query(1, 1, n, 1, n, h) << "\n";
}
}
return 0;
}
Complexity Analysis
Range add: $O(\log N)$ per operation.
Clamp: Amortized $O(\log^2 N)$ per operation (Segment Tree Beats analysis).
Query: $O(\log N)$ per query.
Space: $O(N)$.
Note.
For the full IOI 2005 problem with linear-function additions ($f(x) \gets f(x) + s + d(x - l)$ on $[l, r]$), each node stores a lazy tag $(s, d)$ and tracks min/max at segment endpoints. The clamp logic remains the same.
Code
C++ solution used for this page.
// IOI 2005 - Mountain
// Segment tree with range-add, clamp-to-zero, and leftmost-threshold query.
// Clamp uses segment-tree-beats style: recurse only on mixed-sign segments.
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000005;
struct Node {
long long mn, mx;
long long lazy_add;
bool lazy_set;
long long set_val;
};
Node tree[4 * MAXN];
int n;
void build(int v, int l, int r) {
tree[v] = {0, 0, 0, false, 0};
if (l == r) return;
int mid = (l + r) / 2;
build(2 * v, l, mid);
build(2 * v + 1, mid + 1, r);
}
void pushDown(int v, int l, int r) {
if (l == r) return;
for (int c : {2 * v, 2 * v + 1}) {
if (tree[v].lazy_set) {
tree[c].mn = tree[c].mx = tree[v].set_val;
tree[c].lazy_add = 0;
tree[c].lazy_set = true;
tree[c].set_val = tree[v].set_val;
}
tree[c].mn += tree[v].lazy_add;
tree[c].mx += tree[v].lazy_add;
if (tree[c].lazy_set)
tree[c].set_val += tree[v].lazy_add;
else
tree[c].lazy_add += tree[v].lazy_add;
}
tree[v].lazy_add = 0;
tree[v].lazy_set = false;
}
void pull(int v) {
tree[v].mn = min(tree[2 * v].mn, tree[2 * v + 1].mn);
tree[v].mx = max(tree[2 * v].mx, tree[2 * v + 1].mx);
}
void rangeAdd(int v, int l, int r, int ql, int qr, long long val) {
if (ql > r || qr < l) return;
if (ql <= l && r <= qr) {
tree[v].mn += val;
tree[v].mx += val;
if (tree[v].lazy_set) tree[v].set_val += val;
else tree[v].lazy_add += val;
return;
}
pushDown(v, l, r);
int mid = (l + r) / 2;
rangeAdd(2 * v, l, mid, ql, qr, val);
rangeAdd(2 * v + 1, mid + 1, r, ql, qr, val);
pull(v);
}
void clampZero(int v, int l, int r) {
if (tree[v].mn >= 0) return;
if (tree[v].mx <= 0) {
tree[v].mn = tree[v].mx = 0;
tree[v].lazy_add = 0;
tree[v].lazy_set = true;
tree[v].set_val = 0;
return;
}
if (l == r) {
tree[v].mn = tree[v].mx = max(tree[v].mn, 0LL);
return;
}
pushDown(v, l, r);
int mid = (l + r) / 2;
clampZero(2 * v, l, mid);
clampZero(2 * v + 1, mid + 1, r);
pull(v);
}
int query(int v, int l, int r, int ql, int qr, long long h) {
if (ql > r || qr < l || tree[v].mx < h) return -1;
if (l == r) return l;
pushDown(v, l, r);
int mid = (l + r) / 2;
int res = query(2 * v, l, mid, ql, qr, h);
if (res != -1) return res;
return query(2 * v + 1, mid + 1, r, ql, qr, h);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int Q;
cin >> n >> Q;
build(1, 1, n);
while (Q--) {
char op;
cin >> op;
if (op == 'A') {
int l, r;
long long val;
cin >> l >> r >> val;
rangeAdd(1, 1, n, l, r, val);
} else if (op == 'M') {
clampZero(1, 1, n);
} else { // 'Q'
long long h;
cin >> h;
cout << query(1, 1, n, 1, n, h) << "\n";
}
}
return 0;
}
Source Files and Assets
Raw files are still available here when you want the original TeX, C++, or statement assets.