Update disjoint_set.jl

JuliaCollections · Feb 14, 2025 · 4dd340c · 4dd340c
1 parent 07d5bb1
commit 4dd340c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/disjoint_set.jl b/src/disjoint_set.jl
@@ -32,6 +32,11 @@ end
 IntDisjointSet(n::T) where {T<:Integer} = IntDisjointSet{T}(collect(Base.OneTo(n)), zeros(T, n), n)
 IntDisjointSet{T}(n::Integer) where {T<:Integer} = IntDisjointSet{T}(collect(Base.OneTo(T(n))), zeros(T, T(n)), T(n))
 Base.length(s::IntDisjointSet) = length(s.parents)
+function Base.sizehint!(s::IntDisjointSet, n::Integer)
+    sizehint!(s.parents, n)
+    sizehint!(s.ranks, n)
+    return s
+end
 
 """
     num_groups(s::IntDisjointSet)