union-find code snippets in rust