拆地毯——kruskal最大生成树

P2121 拆地毯 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

 

这道题很简单，就是最大生成树。但是之前接触的都是最小生成树，其实就是把排序变成从大到小排序。因为只留下k条地毯，所以当k条地毯在同一个连通块后，我们就可以输出求和的答案，然后结束。

唯一的亮点就是排序是从大到小。

#include <bits/stdc++.h>
using namespace std;

const int N=2e5+100;
int n,m,k,idx,h[N],p[N];
int ans;

struct node
{
    int a,b,w;
}e[N];


bool cmp(const node&s1,const node&s2)
{
    return s1.w>s2.w;
}

int find(int x)
{
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}

void kruskal()
{
    sort(e+1,e+1+m,cmp);
    for(int i=1;i<=n;i++)p[i]=i;
    
    int q=0;
    for(int i=1;i<=m;i++)
    {
        int a=e[i].a,b=e[i].b,w=e[i].w;
        if(find(a)!=find(b))
        {
            ans+=w;
            q++;
            p[find(a)]=find(b);
        }
        if(q==k)
        {
            printf("%d\n",ans);
            return ;
        }
    }
}


int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        e[i].a=u,e[i].b=v,e[i].w=w;
    }
    
    kruskal();
    
    return 0;
}