牛客网算法总结——并查集（一星题）

背景介绍

本人算法小白在做动态规划时 偶然发现并查集这个算法 所以赶紧过来做了几道一星题 感觉这个算法非常好用并且比较简单 所以在此总结一下 （下面这几道题难度也非常友好 都是一道比一道难一点 并且对理解很有帮助 很适合小白。
https://ac.nowcoder.com/acm/skill/detail/acm/1291

加边的无向图

这是一道很常规的并查集的题 用并查集算法很轻松就找出有几个不相交的集合然后解出

#include <iostream>
#include <set>
using namespace std;
int pre[100005];
int find(int x){
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}
inline void merge(int i, int j)
{
    pre[find(i)] = find(j);
}
int main()
{
    int n,m;
    cin>>n>>m;
    int a,b;
    set<int> s;
    for(int i=0;i<=n;i++)pre[i]=i;
    for(int i=0;i<m;i++){
        cin>>a>>b;
        merge(a,b);
    }
    for(int i=1;i<=n;i++){
        s.insert(find(i));
    }
    cout<<s.size()-1<<endl;
    return 0;
}

Call to your teacher

这道题也是很简单的并查集 在上一道题基础上 注意一个点就可以了（如果没人认识老师就不可能联系到老师 但是如果老师认识别人就会把自己送入这个集合 就会误认为联系得到老师）把老师认识别人这个不加入集合就行。

#include <iostream>
using namespace std;
int pre[55];
int find(int x){
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}
inline void merge(int i, int j)
{
    pre[find(i)] = find(j);
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)pre[i]=i;
    int a,b;
    for(int i=0;i<m;i++){
        cin>>a>>b;
        if(a==n)continue;//将老师认识别人去掉
        merge(a,b);
    }
    if(find(1)==find(n))cout<<"Yes"<<endl;
    else cout<<"No"<<endl;
    return 0;
}

任意点

这道题在并查集基础上加上本题的特色 当两个点的x轴或y轴相等就说明能一次到达 就加入集合就行

#include <iostream>
#include <set>
using namespace std;
struct point{
    int x,y;
};
point points[105];
int pre[105];
set<int> s;
int find(int x){
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}
void merge(int i,int j){
    pre[find(i)]=find(j);
}
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        pre[i]=i;
        cin>>points[i].x>>points[i].y;
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(points[i].x==points[j].x||points[i].y==points[j].y)
                merge(i,j);
        }
    }
    for(int i=0;i<n;i++)s.insert(find(i));
    cout<<s.size()-1<<endl;
    return 0;
}

白色相簿

最后一题也就是这个简单的时间不够 需要加权标记法 使用rank数组减少时间

#include <iostream>
using namespace std;
int pre[10005];
int people[10005];
int ranks[10005];
int find(int x){
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}
void merge(int i, int j)
{
    int x = find(i), y = find(j);
    if (ranks[x] <= ranks[y])
        pre[x] = y;
    else
        pre[y] = x;
    if (ranks[x] == ranks[y] && x != y)
        ranks[y]++;
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        ranks[i]=1;
        pre[i]=i;
    }
    int a,b;
    for(int i=0;i<m;i++){
        cin>>a>>b;
        if(a==b)continue;
        merge(a,b);
    }
    for(int i=1;i<=n;i++){
        int k=find(i);
        people[k]++;
        if(people[k]==3){
            cout<<"Error"<<endl;
            return 0;
        }
    }
    cout<<"Nice"<<endl;
    return 0;
}

总结

最后放出经常用到的几个函数

int find(int x){
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}
void merge(int i, int j)
{
    int x = find(i), y = find(j);
    if (ranks[x] <= ranks[y])
        pre[x] = y;
    else
        pre[y] = x;
    if (ranks[x] == ranks[y] && x != y)
        ranks[y]++;
}