首页 > 其他 > 详细

P2121 拆地毯

时间:2020-05-01 18:27:49      阅读:58      评论:0      收藏:0      [点我收藏+]

算法

最小生成树

思路

就排个序然后做 k 条边的最大生成树

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,k,f[100100],ans;
struct no
{
    int x,y,z;
}a[100100];
bool comp (const no & a,const no &b)
{
    return a.z>b.z;
}
int find(int x)
{
    if(f[x]!=x)
        f[x]=find(f[x]);
    return f[x];
}
void un(int x,int y)
{
    f[x]=y;
}
int main()
{

    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+m+1,comp);
    for(int i=1;i<=n;i++)//并查集,初始时每个祖先都是自己
        f[i]=i;
    for(int i=1,kk=0;i<=m&&kk<k;i++)
    {
        int l=find(a[i].x),r=find(a[i].y);
        if(l!=r)//不在同一集合就合并
        {
            un(l,r);
            kk++;//计数
            ans+=a[i].z;
        }
    }
    printf("%d",ans);
    return 0;
}

  

 

P2121 拆地毯

原文:https://www.cnblogs.com/ruanmowen/p/12813907.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!