disjoint set code snippets in rust