一、并查集是什么?
并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中。其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
二、如何使用
代码如下(Kruskal):
#include<iostream>
#include<cstring>
#include<algorithm>
#define N 1000
using namespace std;
struct Edge{
int a;
int b;
int c;
};
Edge at[N];
int n,m;
int parent[N];
bool cmp(Edge x,Edge y){
return x.c<y.c;
}
int find(int x){
if(parent[x]==x){
return x;
}else{
return parent[x]=find(parent[x]);
}
}
int join(int x,int y){
int find_x=find(x);
int find_y=find(y);
if(find_x!=find_x){
parent[find_x]=find_y;
return 0;
}else{
return 1;
}
}
int main(){
cin>>n>>m;
memset(parent,-1,sizeof(parent));
for(int i=1;i<m;i++){
cin>>at[i].a>>at[i].b>>at[i].c;
}
sort(at,at+m,cmp);
int r=0;
for(int i=0;i<m;i++){
int x=find(at[i].a);
int y=find(at[i].b);
if(x!=y){
parent[x]=y;
r+=at[i].c;
}
}
cout<<r<<endl;
return 0;
}
当然你把join里的判断放到main里也行