洛谷 1195 口袋的天空 最小生成樹 解題報告

題目背景

小杉坐在教室裏，透過口袋一樣的窗戶看口袋一樣的天空。

有很多雲飄在那裏，看起來很漂亮，小杉想摘下那樣美的幾朵雲，做成棉花糖。

題目描述

給你雲朵的個數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

思路

很簡單的，直接最小生成樹

代碼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1000+5; 
int n,m,k,sum,ans,fa[N];
struct edge
{
    int u,v,w;
    bool operator <(edge a) const
    {return w<a.w;}
}ed[N*10];
int find(int x)
{
    if (fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (int i=0;i<=n;i++) 
    fa[i]=i;
    for (int i=1;i<=m;i++)
    scanf("%d%d%d",&ed[i].u,&ed[i].v,&ed[i].w);
    sort(ed+1,ed+m+1);
    for (int i=1;i<=m;i++)
    {
        int fx=find(ed[i].u),fy=find(ed[i].v);
        if (fx!=fy)
        {
            fa[fx]=fy;
            sum++;
            ans+=ed[i].w;
        }
        if (sum==n-k) {printf("%d",ans);return 0;}
    }
    puts("No Answer");   
    return 0;
}