专题四 · 1008

代码及解释

#include<stdio.h>
#include<string.h>
#include<algorithm>

// 并查集的应用
// 判断边的两端是否在同一个联通分量里
// 然后在最后合并成一个回路

struct node {
int u,v,val;
} edge[100005];

int set[10005], circle[10005];

int find_f(int x) {
return set[x]==x ? x : set[x] = find_f(set[x]);
}

bool merge_branch(int x,int y){
int a = find_f(x);
int b = find_f(y);
if (a == b) {
if(circle[a] == -1){
circle[a] = 1;
return true;
}
return false;
} else {
if(circle[a] == circle[b] && circle[a] == 1)
return false;
if(circle[a] == 1)
set[b] = a;
else
set[a] = b;
return true;
}
}

int cmp(node a,node b){
return a.val > b.val;
}

int main() {
int n, m;
while(scanf("%d%d",&n,&m)==2 && n && m){

for(int i=0;i<m;i++)
scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].val);

for(int i = 0; i < n; ++i){
set[i] = i;
circle[i] = -1;
}

std::sort(edge,edge+m,cmp);
int ans = 0;

for(int i=0;i<m;i++){
if(merge_branch(edge[i].u, edge[i].v))
ans += edge[i].val;
}

printf("%d\n",ans);
}
return 0;
}