P2121 拆地毯

\(kruskal\)同样可用于求最大生成树,本题只需求出最大生成树的前\(k\)条边即可

const int N=1e5+10;
struct Node
{
    int a,b,c;
    bool operator<(const Node &W) const
    {
        return c>W.c;
    }
}e[N];
int p[N];
int n,m,k;

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

int kruskal()
{
    int res=0,cnt=0;
    for(int i=0;i<m;i++)
    {
        int a=e[i].a,b=e[i].b,c=e[i].c;
        int pa=find(a),pb=find(b);
        if(pa != pb)
        {
            p[pa]=pb;
            res+=c;
            cnt++;
            if(cnt == k) return res;
        }
    }
}

int main()
{
    cin>>n>>m>>k;

    for(int i=1;i<=n;i++) p[i]=i;

    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        e[i]={a,b,c};
    }

    sort(e,e+m);

    int t=kruskal();
    cout<<t<<endl;
    //system("pause");
}
posted @ 2020-09-07 14:05  Dazzling!  阅读(132)  评论(0编辑  收藏  举报