684. 冗余连接

题目给定一个树添加一条边后的图，需要找出添加的边。思路为遍历所有边，用并查集维护集合，当一条边的两个端点已经在同一集合时，说明即为最后添加的边。

const int N=1010;
int p[N];
class Solution {
public:
    int find(int x){
        while(p[x]!=x){
            x=p[x];
        }
        return x;
    }
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        //并查集
        for(int i=0;i<N;i++)
            p[i]=i;
        for(vector<int> edge:edges){
            int a=edge[0],b=edge[1];
            if(find(a)!=find(b)){
                p[find(b)]=find(a);
            }else{
                return {a,b};
            }
        }
        return {};
    }
};