(phSr"SS5rg)z Disjoint set data structure c^\rSrSrSrSSjrSrSrSrSr S r S r S r S r S rSrSrg) DisjointSetaDisjoint set data structure for incremental connectivity queries. .. versionadded:: 1.6.0 Attributes ---------- n_subsets : int The number of subsets. Methods ------- add merge connected subset subset_size subsets __getitem__ Notes ----- This class implements the disjoint set [1]_, also known as the *union-find* or *merge-find* data structure. The *find* operation (implemented in `__getitem__`) implements the *path halving* variant. The *merge* method implements the *merge by size* variant. References ---------- .. [1] https://en.wikipedia.org/wiki/Disjoint-set_data_structure Examples -------- >>> from scipy.cluster.hierarchy import DisjointSet Initialize a disjoint set: >>> disjoint_set = DisjointSet([1, 2, 3, 'a', 'b']) Merge some subsets: >>> disjoint_set.merge(1, 2) True >>> disjoint_set.merge(3, 'a') True >>> disjoint_set.merge('a', 'b') True >>> disjoint_set.merge('b', 'b') False Find root elements: >>> disjoint_set[2] 1 >>> disjoint_set['b'] 3 Test connectivity: >>> disjoint_set.connected(1, 2) True >>> disjoint_set.connected(1, 'b') False List elements in disjoint set: >>> list(disjoint_set) [1, 2, 3, 'a', 'b'] Get the subset containing 'a': >>> disjoint_set.subset('a') {'a', 3, 'b'} Get the size of the subset containing 'a' (without actually instantiating the subset): >>> disjoint_set.subset_size('a') 3 Get all subsets in the disjoint set: >>> disjoint_set.subsets() [{1, 2}, {'a', 3, 'b'}] NcSUl0Ul0Ul0Ul0UlUbUHnUR U5 M gg)N) n_subsets_sizes_parents_nbrs_indicesadd)selfelementsxs K/var/www/html/venv/lib/python3.13/site-packages/scipy/_lib/_disjoint_set.py__init__DisjointSet.__init__[sE       c,[UR5$)zcReturns an iterator of the elements in the disjoint set. Elements are ordered by insertion order. )iterr r s r__iter__DisjointSet.__iter__gs DMM""rc,[UR5$N)lenr rs r__len__DisjointSet.__len__ns4==!!rcXR;$rr r rs r __contains__DisjointSet.__contains__qsMM!!rcXR;a [U5eURnURUURX!:wa1X"UX!'X!nURUURX!:waM1U$)zFind the root element of `x`. Parameters ---------- x : hashable object Input element. Returns ------- root : hashable object Root element of `x`. )r KeyErrorr )r rparentss r __getitem__DisjointSet.__getitem__tsy MM !1+ --mmA$-- ";; ,GJ AmmA$-- ";;rcXR;agSURU'XRU'XRU'[ UR5URU'U=R S- slg)z(Add element `x` to disjoint set N)r rr r rrr s rr DisjointSet.addsZ   A a 1 t}}- a !rcXnXnURUURU:XagURnXSURU4XTURU4:aXCpCX0RU'URU==URU- ss'URUURUsURU'URU'U=RS-slg)aMerge the subsets of `x` and `y`. The smaller subset (the child) is merged into the larger subset (the parent). If the subsets are of equal size, the root element which was first inserted into the disjoint set is selected as the parent. Parameters ---------- x, y : hashable object Elements to merge. Returns ------- merged : bool True if `x` and `y` were in disjoint sets, False otherwise. Fr)T)r rr r r)r ryxryrsizess rmergeDisjointSet.merges"W W ==  b 1 1  It}}R( )UY b8I,J J b B4;;r?*)-BB& 2 2 !rcHURXURX:H$)zTest whether `x` and `y` are in the same subset. Parameters ---------- x, y : hashable object Elements to test. Returns ------- result : bool True if `x` and `y` are in the same set, False otherwise. r)r rr,s r connectedDisjointSet.connecteds%}}TW%tw)???rc4XR;a [U5eU/nURUnURUURU:waBURU5 URUnURUURU:waMB[ U5$)zGet the subset containing `x`. Parameters ---------- x : hashable object Input element. Returns ------- result : set Subset containing `x`. )r r$r appendset)r rresultnxts rsubsetDisjointSet.subsets MM !1+ jjmmmC DMM!$44 MM# **S/CmmC DMM!$446{rc$URX$)aVGet the size of the subset containing `x`. Note that this method is faster than ``len(self.subset(x))`` because the size is directly read off an internal field, without the need to instantiate the full subset. Parameters ---------- x : hashable object Input element. Returns ------- result : int Size of the subset containing `x`. )rr s r subset_sizeDisjointSet.subset_sizes"{{47##rc/n[5nUH=nX2;dM URU5nURU5 URU5 M? U$)ziGet all the subsets in the disjoint set. Returns ------- result : list Subsets in the disjoint set. )r7r:updater6)r r8visitedrxsets rsubsetsDisjointSet.subsetssL%A{{1~t$ d#   r)r r r rrr)__name__ __module__ __qualname____firstlineno____doc__rrrr!r&r r0r3r:r=rC__static_attributes__rrrrsASh #"". > @.$&rrN)rIrrKrrrLs xxr