目录
这是牛客网上题号为 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<iostream>
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;
}