并查集算法:
并查集主要用于解决一些元素分组的问题。它管理一系列不相交的集合,并支持两种操作:
合并(Union):把两个不相交的集合合并为一个集合。
查询(Find):查询两个元素是否在同一个集合中。
1.初始化
const int N = 5e4 + 10 ;
int fa[N];
......
for(int i=0;i<n;i++) fa[i]=i;
2.查询
int find(int x)
{
return x==fa[x]?fa[x]:fa[x]=find(fa[x]);
}
3.合并
void merge(int i, int j)
{
fa[find(i)] = find(j);
}
例题:
D - 畅通工程
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路?
Input
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数,分别是城镇数目N ( < 1000 )和道路数目M;随后的M行对应M条道路,每行给出一对正整数,分别是该条道路直接连通的两个城镇的编号。为简单起见,城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时,输入结束,该用例不被处理。
output
对每个测试用例,在1行里输出最少还需要建设的道路数目。
#include<iostream>
#include <stdio.h>
#include<stdlib.h>
#include <algorithm>
#define end1 "\n"
using namespace std;
const int N = 5e4 + 10 ;
int n,m,k=1;
int fa[N];
int find(int node)
{
if(fa[node]==-1)
return node;
fa[node]=find(fa[node]);
return fa[node];
}
int main()
{
while(1)
{
cin>>n;
if(n==0) break;
cin>>m;
for(int i=1;i<=n;i++) fa[i]=-1;
while(m--)
{
int p,q;
cin>>p>>q;
if(find(p)!=find(q))
fa[find(q)]=find(p);
}
int sum=0;
for(int i=1;i<=n;i++)
{
if(fa[i]==-1)
sum++;
}
sum--;
cout<<sum<<endl;
k++;
}
return 0;
}