1. 程式人生 > >leetcode684+刪除樹中構成圖的多餘邊,並查集

leetcode684+刪除樹中構成圖的多餘邊,並查集

https://leetcode.com/problems/redundant-connection/description/

class Solution {
public:
    int find(int a, int p[])
    {
        int res = a;
        while(res!=p[res]){
            res = p[res];
        }
        return res;
    }
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        int parent[edges.size()+1];
        for(int i=1; i<=edges.size(); i++){
            parent[i] = i;
        }
        for(auto e:edges){
            if(find(e[0],parent)==find(e[1], parent))//在一個集合裡面了,說明有圈了
                return e;
            parent[find(e[1], parent)] = find(e[0], parent);
        }
    }
};