有没有优化的方案，这个是n^2复杂度超时了

题目代码如上，用的并查集知识，需要进一步优化。有没有人可以指点一下呢？