leetcode684+刪除樹中構成圖的多餘邊,並查集
阿新 • • 發佈:2018-11-01
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); } } };