DisjointSet (a.k.a. union–find data structure or merge–find set) implementation for Python.
The only requirement is having Python 3 installed, you can verify this by running:
$ python --version
Python 3.7.2
pip install disjoint-set
You can verify he package was installed to your current environment by running:
$ pip list | grep disjoint-set
disjoint-set 0.6.1
>>> from disjoint_set import DisjointSet
>>> ds = DisjointSet()
>>> ds.find(1)
1
>>> ds.union(1,2)
>>> ds.find(1)
2
>>> ds.find(2)
2
>>> ds.connected(1,2)
True
>>> ds.connected(1,3)
False
>>> "a" in ds
False
>>> ds.find("a")
'a'
>>> "a" in ds
True
>>> list(ds)
[(1, 2), (2, 2), (3, 3), ('a', 'a')]
>>> list(ds.itersets())
[{1, 2}, {3}, {'a'}]
Feel free to open any issues on github.
This project is licensed under the MIT License - see the LICENSE.md file for details