Fixed erase return iterator.

5cript · Jul 31, 2024 · dfda487 · dfda487
1 parent 639401d
commit dfda487
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/include/interval-tree/interval_tree.hpp b/include/interval-tree/interval_tree.hpp
@@ -885,14 +885,20 @@ namespace lib_interval_tree
             if (!iter.node_)
                 throw std::out_of_range("cannot erase end iterator");
 
-            auto next = iter;
-            ++next;
+            iterator next = iter;
 
             node_type* y;
             if (!iter.node_->left_ || !iter.node_->right_)
+            {
                 y = iter.node_;
+                ++next;
+            }
             else
+            {
                 y = successor(iter.node_);
+                next = iterator{y, this};
+                ++next;
+            }
 
             node_type* x;
             if (y->left_)