union_find::reserve

Z3Prover · Aug 1, 2022 · 5d858da · 5d858da
1 parent e8e64d3
commit 5d858da
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/util/uint_set.h b/src/util/uint_set.h
@@ -192,7 +192,7 @@ class uint_set : unsigned_vector {
             m_set(&s), m_index(at_end?s.get_max_elem():0), m_last(s.get_max_elem()) {
             scan();
             SASSERT(invariant());
-          }
+        }
         unsigned operator*() const { return m_index; }
         bool operator==(iterator const& it) const { return m_index == it.m_index; }
         bool operator!=(iterator const& it) const { return m_index != it.m_index; }

diff --git a/src/util/union_find.h b/src/util/union_find.h
@@ -93,6 +93,11 @@ class union_find {
         return r;
     }
 
+    void reserve(unsigned v) {
+        while (get_num_vars() <= v)
+            mk_var();
+    }
+
     unsigned get_num_vars() const { return m_find.size(); }