题解 | #还是畅通工程#

还是畅通工程

https://www.nowcoder.com/practice/d6bd75dbb36e410995f8673a6a2e2229

#include <iostream>
#include <algorithm>
using namespace std;

const int N=5000;
int p[N];

struct Edge{
    int a,b,w;
    bool operator< (const Edge &t) const{
        return w<t.w;
    }
}edges[N];

int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main() {
    int n;
    while(cin>>n&&n){
        int m=n*(n-1)/2;
        for(int i=0;i<m;i++){
            int a,b,w;
            cin>>a>>b>>w;
            edges[i]={a,b,w};
        }
        sort(edges,edges+m);
        for(int i=1;i<=n;i++) p[i]=i;
        int res=0,cnt=0;
        for(int i=0;i<m;i++){
            int a=edges[i].a,b=edges[i].b,w=edges[i].w;
            a=find(a),b=find(b);
            if(a!=b){
                p[a]=b;
                res+=w;
                cnt++;
            }
        }
        cout<<res<<endl;
    }
    return 0;
}

全部评论

相关推荐

不愿透露姓名的神秘牛友
09-30 19:49
起名星人:蛮离谱的,直接要求转投销售
投递汇川技术等公司10个岗位
点赞 评论 收藏
分享
offer多多的六边形战士很无语:看了你的博客,感觉挺不错的,可以把你的访问量和粉丝数在简历里提一下,闪光点(仅个人意见)
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务