浙江大学机试-还是畅通工程
kruskal算法的使用。
开始每个节点都是独立的根,利用并查集的思想将节点构建成树。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 105;
struct E{
int w;//权值
int a;
int b;
};
E edge[maxn];
bool cmp(E x,E y){
return x.w<y.w;
}
int tree[maxn];//父亲节点
int find_root(int x){
if(tree[x]==-1) return x;
else{
int root = find_root(tree[x]);
tree[x] = root;
return root;
}
}
int main(){
int n;
while(scanf("%d",&n)!=EOF){
if(n==0) break;
memset(tree,-1,sizeof(tree));
int m = n*(n-1)/2;
for(int i=0;i<m;i++){
scanf("%d %d %d",&edge[i].a,&edge[i].b,&edge[i].w);
}
sort(edge,edge+m,cmp);
for(int i=0;i<n;i++){
tree[i] = -1;
}
int ans = 0;//总权值
for(int i=0;i<m;i++){
int a = find_root(edge[i].a);
int b = find_root(edge[i].b);
if(a!=b){
tree[b] = a;
ans += edge[i].w;
}
}
printf("%d\n",ans);
}
return 0;
}