HDU1863-畅通工程

最新推荐文章于 2025-09-07 00:18:35 发布

转载最新推荐文章于 2025-09-07 00:18:35 发布 · 65 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/YogurtShen/archive/2012/08/30/2664518.html

文章标签：

#php

本文介绍了一种实现最小生成树的Kruskal算法，通过C++代码详细展示了算法的具体实现过程，包括边的定义、排序、并查集操作等关键步骤。

http://acm.hdu.edu.cn/showproblem.php?pid=1863

#include<iostream>
#include<cstdio>
#define N 5050
using namespace std;
int fa[N],sum,n,m; 
struct edge 
{
    int s,t,w;
}e[N];
int cmp(const void *a,const void *b) 
{
    return (*(edge *)a).w-(*(edge *)b).w;
}
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int merge(edge e) 
{
    int x,y;
    x=find(e.s);
    y=find(e.t);
    if(x!=y) 
    {
        ++sum;
        fa[x]=y;
        return e.w;
    }
    return 0;
}
int kruskal() 
{
    int i,ans=0;
    for(i=1;i<=m;i++) 
       fa[i]=i;
    qsort(e,n,sizeof(edge),cmp);
    for(i=0;i<n;i++) 
    {
        ans+=merge(e[i]);
        if(sum==m) 
           return ans;
    }
    if(sum<m) 
       return -1;
}
int main(void) 
{
    int i,ans;
    while(scanf("%d%d",&n,&m),n) 
    {
        sum=1;
        for(i=0;i<n;i++) 
           scanf("%d%d%d",&e[i].s,&e[i].t,&e[i].w);
        ans=kruskal();
        if(ans==-1) 
           printf("?\n");
        else 
           printf("%d\n",ans);
    }
    return 0;
}