题解 | 继续畅通工程

继续畅通工程

https://www.nowcoder.com/practice/16212f7d46e44174b5505997ea998538

#include <any>
#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int fa[110];
int find(int x)
{
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}
typedef struct Link{
    int a,b;
    int len;

}L;

L link[100010];

bool cmp(L l1,L l2)
{
   
    return l1.len<l2.len;
}

int main() {
    int n;
    while(cin>>n)
    {
        if(n==0)break;
        for(int i=0;i<=n;i++)fa[i]=i;
        int cnt=0;
        for(int i=0;i<n*(n-1)/2;i++)
        {
            int a,b,len,flag;
            cin>>a>>b>>len>>flag;
            if(a>b)swap(a,b);
            if(flag)
            {
                int f1 = find(a);
                int f2 = find(b);
                if(f1!=f2)
                {
                    fa[f2]=f1;
                }
            }else {
                link[cnt].a=a;
                link[cnt].b=b;
                link[cnt].len=len;
                cnt++;
            }
        }
        sort(link,link+cnt, cmp);
        int ans=0;
        for(int i=0;i<cnt;i++)
        {
            int f1 = find(link[i].a);
            int f2 = find(link[i].b);
            if(f1!=f2)
            {
                fa[f2]=f1;
                ans+=link[i].len;

            }
        }
        cout<<ans<<endl;


    }
}
// 64 位输出请用 printf("%lld")

全部评论

相关推荐

09-25 18:40
已编辑
河北大学 Unity3D客户端
点赞 评论 收藏
分享
真的很糟糕:不一定是你的问题,当然你也可以做的更好一些,继续投相信自己一定会有的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务