shrink_to_fit for mutable graphs

vigna · Feb 1, 2025 · d6c1849 · d6c1849
1 parent de8e1fa
commit d6c1849
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/src/graphs/btree_graph.rs b/src/graphs/btree_graph.rs
@@ -176,6 +176,16 @@ impl<L: Clone + 'static> LabeledBTreeGraph<L> {
         g.add_arcs(arcs);
         g
     }
+
+    /// Shrink the capacity of the graph to fit its current size.
+    ///
+    /// # Implementation Notes
+    ///
+    /// This method just shrinks the capacity of the successor vector, as
+    /// [`BTreeSet`] does not have a `shrink_to_fit` method.
+    pub fn shrink_to_fit(&mut self) {
+        self.succ.shrink_to_fit();
+    }
 }
 
 impl<L: Clone + 'static> SequentialLabeling for LabeledBTreeGraph<L> {
@@ -314,6 +324,16 @@ impl BTreeGraph {
         g.add_arcs(arcs);
         g
     }
+
+    /// Shrink the capacity of the graph to fit its current size.
+    ///
+    /// # Implementation Notes
+    ///
+    /// This method just shrinks the capacity of the successor vector, as
+    /// [`BTreeSet`] does not have a `shrink_to_fit` method.
+    pub fn shrink_to_fit(&mut self) {
+        self.0.shrink_to_fit();
+    }
 }
 
 impl<'a> IntoLender for &'a BTreeGraph {

diff --git a/src/graphs/vec_graph.rs b/src/graphs/vec_graph.rs
@@ -200,6 +200,14 @@ impl<L: Clone + 'static> LabeledVecGraph<L> {
         g.add_arcs(arcs);
         g
     }
+
+    /// Shrink the capacity of the graph to fit its current size.
+    pub fn shrink_to_fit(&mut self) {
+        self.succ.shrink_to_fit();
+        for s in self.succ.iter_mut() {
+            s.shrink_to_fit();
+        }
+    }
 }
 
 impl<L: Clone + 'static> SequentialLabeling for LabeledVecGraph<L> {
@@ -388,6 +396,11 @@ impl VecGraph {
         g.add_arcs(arcs);
         g
     }
+
+    /// Shrink the capacity of the graph to fit its current size.
+    pub fn shrink_to_fit(&mut self) {
+        self.0.shrink_to_fit();
+    }
 }
 
 impl<'a> IntoLender for &'a VecGraph {