洛谷 P1195 口袋的天空

P1195 口袋的天空

题目背景

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

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

题目描述

给你云朵的个数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

说明

厦门一中YMS原创

/*
最小生成树 
跑最小生成树的过程中
连完n-k个边时结束 
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxx 1010
#define maxn 10010
using namespace std;
int n,m,w;
int fa[maxx];
struct edge
{
    int from;
    int to;
    int val;
}e[maxn];
int cmp(edge a,edge b)
{
    return a.val<b.val;
}
int find(int x)
{
    if(x!=fa[x])fa[x]=find(fa[x]);
    return fa[x];
}
void unionn(int x,int y)
{
    int a=find(x);
    int b=find(y);
    fa[b]=a;
}
int main()
{
    scanf("%d%d%d",&n,&m,&w);
    for(int i=1;i<=m;i++)
      scanf("%d%d%d",&e[i].from,&e[i].to,&e[i].val);
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=n;i++)fa[i]=i;
    int sum=0,ans=0;
    for(int i=1;i<=m;i++)
    {
        int x=e[i].from;
        int y=e[i].to;
        if(find(x)!=find(y))
        {
            unionn(x,y);
            sum++;
            ans+=e[i].val;
        }
        if(sum==n-w)break;
    }
    if(sum==n-w) printf("%d
",ans);
    else printf("No Answer
");
    return 0;
}