依旧是并查集问题,这道题目正好给定顶点数目和边的数目相等,只要找到其中的一条边删除将图转化为树就行,而这个多余的边起始就是并查集的添加过程中二者是同一个根(两个顶点早已经联通了),这时直接返回这条边就行
注意这题中的顶点编号从1开始,为保证一致性,father数组大小为n+1,0索引不使用
class Solution {
private:vector<int> father;void init() {for(int i=0;i<father.size();i++)father[i]=i;}int find(int v) {return v==father[v]?v:father[v]=find(father[v]);}bool isSame(int u,int v) {u=find(u);v=find(v);return u==v;}bool join(int u,int v) {u=find(u);v=find(v);if(u==v)return false;else{father[v]=u;return true;}}
public:vector<int> findRedundantConnection(vector<vector<int>>& edges) {int n=edges.size();this->father=vector<int>(n+1);this->init();for(int i=0;i<n;i++){int u=edges[i][0];int v=edges[i][1];if(!this->join(u,v))return {u,v};}return {};}
};