题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1863
kruskal算法思想:目标:得到最小生成树。步骤:把边分为两个集合,每次从集合中取出 权值最小的未成树的边 加入到成树的集合里,当有n-1条边时得到最小生成树。
hdu-1863 畅通工程(kruskal算法+并查集)
#include<cstdio>
#include<cstdlib>
#define MAXN 10000+10
using namespace std;
//par数组用于表示并查集
int par[MAXN],Rank[MAXN];
//a结构体 表示每条边
typedef struct{
int a,b,price;
}Node;
Node a[MAXN];
//用于qsort函数,返回值 a>b +,a<b -,a==b 0;还需要强制类型转换,返回值为int
int cmp(const void*a,const void *b){
return ((Node*)a)->price - ((Node*)b)->price;
}
void Init(int n){
for(int i=0;i<n;i++){
Rank[i]=0;
par[i]=i;//初始化把每个节点的父节点初始化为自己
}
}
int find(int x){//找到编号为x的点的父亲节点
int root = x;
while(root != par[root])root = par[root];//找到最父亲的节点??如果两个点互相死循环怎么办?
while(x != root){//如果最父亲节点不是它自己,把它的父节点设为最父亲节点,然后继续把它父亲的父亲节点设为最父亲节点,直到 全部父亲节点都为最父亲节点
int t = par[x];
par[x] = root;
x = t;
}
return root;//返回最父亲节点
}
void unite(int x,int y){
x = find(x);
y = find(y);
if(Rank[x]<Rank[y]){
par[x]=y;
}
else{
par[y]=x;
if(Rank[x] == Rank[y]) Rank[x]++;
}
}
//n为边的数量,m为村庄的数量
int Kruskal(int n,int m){
//nEdge是边的编号,res是最小代价和
int nEdge = 0,res=0;
//将边按照权值从小到大排序
qsort(a,n,sizeof(a[0]),cmp); //!!!注意,一定是某一个元素的大小,不能是a,a只是指针大小,如果遇到不是int,就会出错
for(int i=0;i<n&&nEdge!=m-1;i++){
//判断当前这条边的两个端点是否属于同一棵树的并查集
//如果不在 就加到一起,然后算出当前最小代价和,指向编号为下一个的边
if(find(a[i].a)!=find(a[i].b)) {
unite(a[i].a,a[i].b);
res += a[i].price;
nEdge++;
}
}
//如果加入边的数量小于m-1,则该无向图不连通,等价于不存在最小生成树
if(nEdge < m-1) res = -1;
return res;
}
int main(){
int n,m,ans;
while(scanf("%d%d",&n,&m)&&n!=0){
Init(m);
for(int i=0;i<n;i++){
scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].price);
//将点编号变为0~m-1
a[i].a--;
a[i].b--;
}
ans = Kruskal(n,m);
if(ans == -1)printf("?\n");
else printf("%d\n",ans);
}
return 0;
}