代码随想录算法训练
—day62
文章目录
前言
今天是算法营的第62天,希望自己能够坚持下来!
今天继续并查集系列!今日任务:
● 108.冗余连接
● 109.冗余连接II
一、108.冗余连接
思路:
无向图,返回一条可以删去的边,使得结果图是一个有着N个节点的树。
如果有多个答案,则返回二维数组中最后出现的边。
那么我们就可以从前向后遍历每一条边(因为优先让前面的边连上),边的两个节点如果不在同一个集合,就加入集合。
如果边的两个节点已经出现在同一个集合里,说明着边的两个节点已经连在一起了,再加入这条边一定就出现环了。
代码如下:
#include<iostream>
#include<vector>
using namespace std;
int n;
vector<int> father = vector<int>(1001, 0);
void init() {
for (int i = 1; i <= n; i++) {
father[i] = i;
}
}
int find(int u) {
return u == father[u]? u : father[u] = find(father[u]);
}
bool isSame(int u, int v) {
u = find(u);
v = find(v);
return u == v;
}
void join(int u, int v) {
u = find(u);
v = find(v);
if (u == v) return;
father