利用并查集计算图的连通分支
题目描述
输入描述:
每个输入文件包含若干行,每行两个整数i,j,表示节点i和j之间存在一条边。
输出描述:
输出每个图的连通分支数。
示例
输入:
1 4
4 3
5 5
输出 :
2
#include <iostream>
#include <map>
using namespace std;
map<int,int> graph; //前者为节点,后者为其对应的最近父节点
int find(int x) //用于查找x的根父节点
{
if(graph.count(x)==0) //不存在x
graph[x] = x;
if (graph[x]==x)
return x;
else
return find(graph[x]);
}
int main()
{
int res = 0;
int x,y;
while(cin>>x>>y) //将y作为x的父节点
{
int fy = find(y);
int fx = find(x);
if(fx!=fy)
graph[fx]=fy;
}
for (auto it=graph.begin();it!=graph.end();++it)
{
if (it->second==it->first)
res++;
}
cout<<res<<endl;
return 0;
}
题解:
利用并查集的思想,若集合A中的某一个元素x属于集合B,则将集合A其余部分作为集合B中x的子集合。此与连通的概念大同小异,当边<i,j>中i位于集合A、j位于集合B时,则可将A全部并入B中.
- 1.并查集一般使用数组实现,数组中存放的元素即为下标对应的父节点。由于节点个数没有说明,使用map结构可以节约不少空间。
- 2.map的key值表示节点,value值表示此节点所对应的父节点。
- 3.初始化:对于map中不存在的元素,将其value值赋值为key值,表示其为一个孤立的点。
- 4.对于输入边<i,j>,点i与点j必定连通,令点i的根父节点作为点j的根父节点的子节点(即可将i所在的集合全部并入j所在的集合)。若i与j本就属于同一个集合则不需要进行操作。
- 5.在计算连通分支数时只需统计根节点的个数即可。(当key==value时表示此节点的父节点是本身,已经没有父节点了,即为根节点)