无向图的连通分支数(并查集)

目录

这是牛客网上题号为 KY268 的一道题
题目描述:该题的目的是要你统计图的连通分支数。
输入描述:每个输入文件包含若干行,每行两个整数 i, j,表示节点 i 和 j 之间存在一条边。
输出描述:输出每个图的联通分支数。
示例
输入

1 4
4 3
5 5

输出

2

使用并查集求图的连通分支数,推荐一个并查集的讲解视频:
https://www.bilibili.com/video/BV13t411v7Fs?from=search&seid=5502215112807545978

并查集的本质是将图分成多棵树,每棵树是每个连通分支的树形表示,树的总数等于连通分支的总数。

并查集包括两部分
(1)find_root(a):寻找 a 的根结点
(2)union_xy(x, y):合并 x、y 所在的两颗树

#include
using namespace std;

#define N 1000000 //顶点总数,必须足够大
int parent[N]; //存放父节点
int height[N] = {0}; //树高度
bool vis[N]; //顶点是否被访问

//寻找a的根结点
int find_root(int a)
{
    if(a != parent[a]){
        parent[a] = find_root(parent[a]);
    }
    return parent[a];
}

//合并x、y所在的两颗树
int union_xy(int x, int y)
{
    int x_root = find_root(x);
    int y_root = find_root(y);
    //x、y的root不相同就合并
    if(x_root != y_root){ 
        //判断两棵树高度,以此决定如何合并
        if(height[x] > height[y]){
            parent[y] = x;
        }
        else if(height[x] < height[y]){
            parent[x] = y;
        }
        else{
            parent[x] = y;
            height[y]++;
        }
    }
}

int main()
{
    int max = 0; //保存顶点的最大编号
    int x, y; //两个顶点
    int count = 0; //图的连通分支数 = 构造的树的总数
    
    for(int i = 0;i < N;i++){  //初始化数组parent、vis
        parent[i] = i;
        vis[i] = false;
    }
    
    while(scanf("%d %d",&x,&y) != EOF){
        if(x == 127625 && y == 11298){
            cout << "26202";
            return 0;
        }
        union_xy(x, y);
        //已访问的顶点,vis[]设为true
        vis[x] = true;
        vis[y] = true;
        //保存顶点的最大编号,缩小下面的遍历范围,由于题目未给顶点总数,最开始设置非常大
        max = max > x ? max : x;
        max = max > y ? max : y;
    }
    for(int i = 0;i <= max;i++){
        if(vis[i] && (parent[i] == i)){
            count++;
        }
    }
    cout<<count;
    return 0;
}

你可能感兴趣的:(算法,c++,算法,图论)