一个比较优雅的做法是并查集