Merge pull request #921 from amontoison/patch-1

Update disjoint_set.jl
JuliaCollections · Feb 20, 2025 · 1c6869c · 1c6869c
2 parents 2126bb5 + 4dd340c
commit 1c6869c
Showing 1 changed file with 6 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)
@@ -182,6 +187,7 @@ 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, n)
     return s
 end