Improve segment_tree

indy256 · Oct 8, 2023 · abf464e · abf464e
1 parent e952b16
commit abf464e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/cpp/structures/segment_tree.h b/cpp/structures/segment_tree.h
@@ -9,13 +9,20 @@ struct segtree {
         long long sum = 0;
         long long add = 0;
 
+        // set initial value for a leave
+        void initialize(long long v) {
+            mx = v;
+        }
+
+        // apply aggregate operation to the node
         void apply(int l, int r, long long v) {
             mx += v;
             sum += (r - l + 1) * v;
             add += v;
         }
     };
 
+    // construct a node from its children
     static node unite(const node &a, const node &b) {
         node res;
         res.mx = max(a.mx, b.mx);
@@ -52,7 +59,7 @@ struct segtree {
     template <class T>
     void build(int x, int l, int r, const vector<T> &v) {
         if (l == r) {
-            tree[x].apply(l, r, v[l]);
+            tree[x].initialize([l]);
             return;
         }
         int m = (l + r) >> 1;