问题 H: 口袋的天空(Kruscal)

题目描述

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

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

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

样例输入复制

3 1 2
1 2 1

样例输出复制

1

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 10005;

struct Edge
{
    int x, y, w;
    bool operator < (const Edge &e) const
    {
        return w < e.w;
    }
}edge[maxn];

int F[maxn];
int num=0,ans=0;
int Find(int x)
{//查找x的父节点
    if (F[x] == x)return x;
    else return F[x] = Find(F[x]);//递归查找,路径压缩
}

int Union(int x, int y,int z)
{
    int dx = Find(F[x]);
    int dy = Find(F[y]);
    if (dx != dy)
    {
        F[dx] = dy;
        num++;
        ans+=z;
    }
    return num;
}

int main()
{
    int N, M, K;

    while(cin >> N >> M >> K)
    {
        ans=0,num=0;
        memset(F, 0, sizeof(F));
        for (int i = 1; i <= N; i++)F[i] = i;
        for (int i = 1;i <= M; i++)cin>>edge[i].x>>edge[i].y>>edge[i].w;
        sort(edge + 1, edge + M + 1);
        bool flag = false;
        for (int i = 1; i <= M; i++)
        {
            if(N- Union(edge[i].x,edge[i].y,edge[i].w)==K)
            {
                flag = true;
                break;
            }
        }
        if(flag)cout<<ans<<endl;
        else cout<<"No Answer"<<endl;
    }
    return 0;
}