Fix BinaryHeap

vadyushkins · Feb 15, 2024 · 9cb6aea · 9cb6aea
1 parent d0bb192
commit 9cb6aea
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/algorithms/binary_heap/binary_heap.hpp b/algorithms/binary_heap/binary_heap.hpp
@@ -91,20 +91,22 @@ struct BinaryHeap {
     }
 
     void remove(size_t key) {
-        std::swap(
-            this->container->begin()[key],
-            this->container->begin()[this->container->size() - 1]
-        );
-        this->container->pop_back();
-        this->sift_down(key);
+        if (this->container->size() > key) {
+            std::swap(
+                this->container->begin()[key],
+                this->container->begin()[this->container->size() - 1]
+            );
+            this->container->pop_back();
+            this->sift_down(key);
+        }
     }
 
     void pop() {
         this->del(0);
     }
 
     void change(size_t key, T new_value) {
-        if (this->container->begin()[key] != new_value) {
+        if (this->container->size() > key && this->container->begin()[key] != new_value) {
             this->container->begin()[key] = new_value;
             if (
                 key > 0 &&