Update find_union_tree.py

h field is needed only in the root of the tree and can be removed from other nodes

Author: beling

find_union_tree.py

@@ -44,8 +44,10 @@ def union(self, other: 'FindUnionTree') -> None:
             other_root.parent = self_root
             if self_root.h == other_root.h:
                 self_root.h += 1
+            del other_root.h
         else:
             self_root.parent = other_root
+            del self_root.h
 
 
 if __name__ == "__main__":	# demo program: