theodesp / unionfind

An idiomatic implementation of a weighted Union Find data structure with path compression in Go.

unionfind

An idiomatic implementation of a weighted Union Find data structure with path compression in go.

Install

$ go get github.com/theodesp/unionfind

Usage

// Initialize with size
uf := unionfind.New(10)

// Union a,b connects components at index a and b
uf.Union(1, 2)
uf.Union(2, 3)
uf.Union(5, 6)
uf.Union(4, 6)

fmt.PrintLn(uf.Find(2)) // Prints 1
fmt.PrintLn(uf.Connected(1, 2)) // Prints true
fmt.PrintLn(uf.Connected(1, 3)) // Prints false

API

uf := unionfind.New(N)

Create a new Union Find Structure with size N.

uf.Union(p, q)

Connects p and q components.

uf.Find(p)

Attempts to find the Root component of p. Returns -1 if p is out of bounds.

uf.Root(p, q)

Same as Find.

uf.Connected(p, q)

Checks if p and q are connected.

EXTRA

There is also a goroutine safe version of unionfind in the file safe-unionfind.

Licence

MIT - Theo Despoudis

unionfind's Issues

Would it be possible to expose the struct field .root?

Howdy - experimenting with this package and have a need to use the root array, which appears to be inaccessible if I call uf.root. Would it be straightforward to implement this, or provide a pointer on how I might implement?

Make threadSafeUnionFind public

Both threadSafeUnionFind and threadSafeUnionFind's constructor are private, so can't be used. If you capitalize to ThreadSafeUnionFind or created a public constructor, then we could use it.

insert?

how does one insert elements? What types are insertable?

i.e., compare to add in another implementation (python): https://raw.githubusercontent.com/deehzee/unionfind/master/unionfind.py