P1195 口袋的天空 洛谷

https://www.luogu.org/problem/show?pid=1195

题目背景

小杉坐在教室里，透过口袋一样的窗户看口袋一样的天空。

有很多云飘在那里，看起来很漂亮，小杉想摘下那样美的几朵云，做成棉花糖。

题目描述

给你云朵的个数N，再给你M个关系，表示哪些云朵可以连在一起。

现在小杉要把所有云朵连成K个棉花糖，一个棉花糖最少要用掉一朵云，小杉想知道他怎么连，花费的代价最小。

输入输出格式

输入格式：

每组测试数据的

第一行有三个数N,M,K(1<=N<=1000,1<=M<=10000,1<=K<=10)

接下来M个数每行三个数X,Y,L，表示X云和Y云可以通过L的代价连在一起。(1<=X,Y<=N,0<=L<10000)

30%的数据N<=100,M<=1000

输出格式：

对每组数据输出一行，仅有一个整数，表示最小的代价。

如果怎么连都连不出K个棉花糖，请输出'No Answer'。

输入输出样例

输入样例#1：

3 1 2
1 2 1

输出样例#1：

1

#include <algorithm>
#include <iostream>

using namespace std;

int n,m,k,tot,ans;
int fa[10015];
struct node
{
    int x,y,l;
}w[10015];

bool cmp(node a,node b)
{
    return a.l<b.l;
}

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

int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
        cin>>w[i].x>>w[i].y>>w[i].l;
    sort(w+1,w+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int xx=find(w[i].x),yy=find(w[i].y);
        if(xx!=yy)
        {
            fa[xx]=yy;
            ans+=w[i].l;
            tot++;
        }
        if(tot==n-k)
        {
            cout<<ans;
            return 0;
        }
    }
    cout<<"No Answer";
    return 0;
}

没过但不明白的一个

#include <algorithm>
#include <iostream>

using namespace std;

int n,m,k,tot,ans;
int fa[10015];
struct node
{
    int x,y,l;
}w[10015];

bool cmp(node a,node b)
{
    return a.l<b.l;
}

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

int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
        cin>>w[i].x>>w[i].y>>w[i].l;
    sort(w+1,w+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int xx=find(w[i].x),yy=find(w[i].y);
        if(xx!=yy)
        {
            fa[xx]=yy;
            ans+=w[i].l;
        }
        for(int j=1;j<=n;j++)
            if(j==find(j))
                tot++;
        if(tot==k)
        {
            cout<<ans;
            return 0;
        }
    }
    
    if(tot!=k)
        cout<<"No Answer";
    return 0;
}