算法学习记录--并查集_并查集优劣-CSDN博客

本文链接：https://blog.csdn.net/Chmaz/article/details/114979807

上大物学了会并查集,感觉挺简单的,而且很好玩,继dfs,bfs和Floyd算法外又学了一种求连通块的算法,综合下来这几种算法各有优劣吧.
并查集算法详解见此
模板如下:

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100001
int father[MAXN],rank_[MAXN];

void init(int n){//初始化
    for(int i=0;i<n;i++){
        father[i]=i;
        rank_[i]=1;
    }
}

int find_(int x){//找到x的根节点,同时把从x到根节点的father都设为根节点--路径压缩
    // if(x!=father[x]){
    //     father[x]=find_(father[x]);
    // }
    return father[x]?x:(father[x]=find_(father[x]));
}

void merge_(int i,int j){//按秩合并--深度更小的树合并到深度更深的树
    int x=find_(i),y=find_(j);
    if(rank_[x]<=rank_[y]){
        father[x]=y;
    }
    else{
        father[y]=x;
    }
    if(rank_[x]==rank_[y]&&x!=y) rank_[y]++;
}

int main(){
    

    return 0;
}

例题1
洛谷P1551 亲戚

题目背景
若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。
*
题目描述规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。
*
输入格式第一行：三个整数n,m,p，（n<=5000,m<=5000,p<=5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。
*
以下m行：每行两个数Mi，Mj，1<=Mi，Mj<=N，表示Mi和Mj具有亲戚关系。*

接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。
*
输出格式 P行，每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
*
输入输出样例输入 #1
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

输出 #1
Yes
Yes
No

这道题非常简单,只需要简单的应用即可
代码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 50010
int father[MAXN],rank_[MAXN];
void init(int n){
    for(int i=0;i<n;i++){
        father[i]=i;
        rank_[i]=1;
    }
}

int find_(int x){
    // if(x!=father[x]){
    //     father[x]=find_(father[x]);
    // }
    return x==father[x]?x:(father[x]=find_(father[x]));
}

void merge_(int i,int j){
    int x=find_(i),y=find_(j);
    if(rank_[x]<=rank_[y]){
        father[x]=y;
    }
    else{
        father[y]=x;
    }
    if(rank_[x]==rank_[y]&&x!=y) rank_[y]++;
}

int main(){
    int n,m,p,x,y;
    cin>>n>>m>>p;
    init(n);
    for(int i=0;i<m;i++){
        scanf("%d%d",&x,&y);
        merge_(x,y);
    }
    for(int i=0;i<p;i++){
        scanf("%d%d",&x,&y);
        if(find_(x)==find_(y)) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}

例题2
洛谷P1551 亲戚

题目背景
若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。
*
题目描述规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。
*
输入格式第一行：三个整数n,m,p，（n<=5000,m<=5000,p<=5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。
*
以下m行：每行两个数Mi，Mj，1<=Mi，Mj<=N，表示Mi和Mj具有亲戚关系。
*
接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。
*
输出格式 P行，每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
*
输入输出样例输入 #1
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
输出 #1
Yes
Yes
No
代码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 1010
int t;
int father[MAXN],rank_[MAXN];
int n,h,r,x[MAXN],y[MAXN],z[MAXN],sp[MAXN];
void init(int n){
    for(int i=0;i<n;i++){
        father[i]=i;
        rank_[i]=1;
    }
    memset(x,0,sizeof(x));
    memset(y,0,sizeof(y));
    memset(z,0,sizeof(z));
    memset(sp,0,sizeof(sp));
}

int find_(int x){
    return x==father[x]?x:(father[x]=find_(father[x]));
}

void merge_(int i,int j){
    int x=find_(i),y=find_(j);
    if(rank_[x]<=rank_[y]){
        father[x]=y;
    }
    else{
        father[y]=x;
    }
    if(rank_[x]==rank_[y]&&x!=y) rank_[y]++;
}
bool is(int i,int j){
    double dis=sqrt((double)(x[i]-x[j])*(x[i]-x[j])+(double)(y[i]-y[j])*(y[i]-y[j])+(double)(z[i]-z[j])*(z[i]-z[j]));
    return 2*r>=dis;
}
int main(){
    
    cin>>t;
    while(t--){
        cin>>n>>h>>r;
        init(n);
        for(int i=0;i<n;i++){
            cin>>x[i]>>y[i]>>z[i];
            if(z[i]-r<=0) sp[i]=-1;
            if(z[i]+r>=h) sp[i]=1;
            if(z[i]-r<=0&&z[i]+r>=h){
                sp[i]=666;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(is(i,j)){
                    if(sp[i]==666){
                        cout<<"Yes"<<endl;
                        goto ddd;
                    }
                    merge_(i,j);
                }
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(sp[i]==-1&&sp[j]==1&&find_(i)==find_(j)){
                    cout<<"Yes"<<endl;
                    goto ddd;
                }
            }
        }
        cout<<"No"<<endl;
        ddd:
        memset(sp,0,sizeof(sp));
    }
    return 0;
}