kstd/vector: implement range erase

1 files changed, 22 insertions, 0 deletions

diff --git a/libs/kstd/include/kstd/vector b/libs/kstd/include/kstd/vector
index 771fc87..9e41cb6 100644
--- a/libs/kstd/include/kstd/vector
+++ b/libs/kstd/include/kstd/vector
@@ -711,6 +711,28 @@ namespace kstd
       return begin() + prefix_size;
     }
 
+    //! Erase a range of elements from this vector.
+    //!
+    //! @param first The start of the range to erase.
+    //! @param last The end of the range to erase.
+    //! @return An iterator pointing to the element after the last deleted element.
+    constexpr auto erase(const_iterator first, const_iterator last) -> iterator
+    {
+      if (first == last)
+      {
+        return begin() + std::ranges::distance(cbegin(), first);
+      }
+
+      auto prefix_size = std::ranges::distance(cbegin(), first);
+      auto element_count = std::ranges::distance(first, last);
+
+      std::ranges::move(begin() + prefix_size + element_count, end(), begin() + prefix_size);
+      destroy_n(end() - element_count, element_count);
+      m_size -= element_count;
+
+      return begin() + prefix_size;
+    }
+
     //! Append a given element to this vector via copy construction.
     constexpr auto push_back(value_type const & value) -> void
     {