1.并查集理论基础
并查集理论基础 | 代码随想录
并查集可以解决什么问题呢?
主要就是集合问题,两个节点在不在一个集合,也可以将两个节点添加到一个集合中。
注意:求根是求箭头出发的数
路径压缩:求根的根。把根的根的....的根,直接作为自己的根
2.寻找存在的路径
107. 寻找存在的路径 | 代码随想录
代码:
#include <iostream>
#include <vector>
using namespace std;
int n; // 节点数量
vector<int> father = vector<int>(101,0); // 按照结点大小定义数组大小
void init(){
for(int i = 0; i < n; i++){
father[i] = i;
}
}
int find(int u){
if(father[u] == u){
return u;
}else{
return 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[v] = u;
}
int main(){
int m,s,t,source,destination;
cin >> n >> m;
init();
while(m--){
cin >> s >> t;
join(s,t);
}
cin >> source >> destination;
if(IsSame(source,destination)) cout << 1 << endl;
else cout << 0 << endl;
}
思路:直接套模板就好了。
我这里一直写错find函数。。。find函数是有函数值的,如果它的根就是自己,那可以直接返回,如果没有,那就返回find(father[u])