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个棉花糖，意味着还有n-k个云朵连在了一起，要使代价最小，把代价最小的k个云朵连在一起即可，当然不能成环

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

const int maxn = 5e3 + 10;
const int maxm = 2e5 + 10;

struct Edge
{
    int u, v, weight;
}edge[maxm];

int n, m, k;
int father[maxn];

int find(int x)
{
    while(x != father[x])
        x = father[x] = father[father[x]];
    return x;
}

bool cmp(const Edge a, const Edge b)
{
    return a.weight < b.weight;
}

int kruskal()
{
    sort(edge + 1, edge + m + 1, cmp);
	int ans = 0;
    int tot = 0;


    for(int i = 1; i <= m; i++){

        int au, av;
        au = find(edge[i].u);
        av = find(edge[i].v);
        if(au == av) //连通
        {
            continue;
        }
		tot++;
        ans += edge[i].weight;
        father[av] = au;//连上去
        if(tot == n - k)break;
    }
    return ans;
}

void slove()
{
    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);
        edge[i].u = u;
        edge[i].v = v;
        edge[i].weight = w;
    }
    for(int i = 1; i <= n; i++)
    {
        father[i] = i;
    }
}

int main()
{
    slove();
    printf("%d",kruskal());
    return 0;
}
/*
3 1 2
1 2 1
*/