口袋的天空洛谷并查集

题目背景

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

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

题目描述

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

又是一道水水的模板题，并查集，排序

//Gang
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#define FOR(x,y,z) for(int x=y;x<=z;x++)
#define REP(x,y,z) for(int x=y;x>=z;x--)
#define ll long long
using namespace std;
int n,m,k;
int tot;
int fa[1000005];
int num=0;
struct node
{
    int x,y,l;
} e[100005];
int cmp(node a,node b)
{
    return a.l<b.l;
}
int find(int x)
{
//    int p,tmp;
//    p=x;
//    while(fa[x]!=x)
//    {
//        x=fa[x];//返回根节点；
//    }
//    while(x!=p)  //路径压缩
//    {
//        tmp = fa[p];
//        fa[p] = x;
//        p = tmp;
//    }
//    return x;//非递归
    return x==fa[x]?x:fa[x]=find(fa[x]);//递归，推荐非递归，递归会暴栈
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    FOR(i,1,m)
    scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].l);
    sort(e+1,e+m+1,cmp);
    FOR(i,1,n)fa[i]=i;
    FOR(i,1,m)
    {
        int fx=find(e[i].x),fy=find(e[i].y);
        if(fx!=fy)
        {
            num++;
            fa[fx]=fy;
            tot+=e[i].l;
        }
        if(num==n-k)
        {
            printf("%d\n",tot);
            return 0;
        }
    }
    printf("No Answer\n");

    return 0;
}