并查集可以的，不过并查集建立好后需要再判断有多少个集合，每个集合多少个元素就好了