Update disjoint_set.jl

Improve `sizehint!(DisjointSets)` such that we preallocate the vectors needed by the internal `IntDisjointSets`.
JuliaCollections · Feb 12, 2025 · 5c7df3d · 5c7df3d
1 parent 77e5ea0
commit 5c7df3d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/disjoint_set.jl b/src/disjoint_set.jl
@@ -182,6 +182,8 @@ Base.empty(s::DisjointSet{T}, ::Type{U}=T) where {T,U} = DisjointSet{U}()
 function Base.sizehint!(s::DisjointSet, n::Integer)
     sizehint!(s.intmap, n)
     sizehint!(s.revmap, n)
+    sizehint!(s.internal.parents, n)
+    sizehint!(s.internal.ranks, n)
     return s
 end