洛谷 1195 口袋的天空

题目背景 Background
小杉坐在教室里,透过口袋一样的窗户看口袋一样的天空。 有很多云飘在那里,看起来很漂亮,小杉想摘下那样美的几朵云,做成棉花糖。

题目描述 Description
给你云朵的个数N,再给你M个关系,表示哪些云朵可以连在一起。
现在小杉要把所有云朵连成K个棉花糖,一个棉花糖最少要用掉一朵云,小杉想知道他怎么连,花费的代价最小。

输入输出格式 Input/output
输入格式:
每组测试数据的
第一行有三个数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’。

输入样例:

3 1 2
1 2 1

输出样例:

1

大体意思是给一个图,生成K个森林,可以通过稍稍改变Kruskal算法来实现生成,并查集初始化后是有N个森林,每次合并两个集合只会令森林数减一,生成K个森林只需要合并N-K次即可。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int size = 200010;
struct dc
{
    int f,t,d;
}l[size];
int f[size];
int find(int x)
{
    if(f[x] == x)
        return x;
    return f[x] = find(f[x]);
}
bool cmp(dc a,dc b)
{
    return a.d < b.d;
}
int main()
{
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    k = n-k;
    int ans = 0;
    for(int i = 1 ; i <= m ; i ++)
        scanf("%d%d%d",&l[i].f,&l[i].t,&l[i].d);
    sort(l+1,l+m+1,cmp);
    for(int i = 1 ; i <= n ; f[i] = i , i ++);
    for(int i = 1 ; i <= m ; i ++)
    {
        int x = find(l[i].f);
        int y = find(l[i].t);
        if(x != y)
        {
            k --;
            f[x] = y;
            ans += l[i].d;
        }
        if(k == 0)
            break;
    }
    if(k)
        puts("No Answer");
    else
        cout<<ans;
    return 0;
}

传送门:洛谷 1995 口袋的天空

你可能感兴趣的:(洛谷 1195 口袋的天空)