[是题解哦] 洛谷 P1195 口袋的天空

题目链接

点这里ww

题目背景

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

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

题目描述

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

题解

这道题明显就是求最小生成树嘛

一开始我理解成了求k棵最小生成树

后来仔细读题

一个棉花糖最少要用掉一朵云

发现是求一棵有n-k条边的最小生成树

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;

const int maxn=1005;
const int maxm=10005;
struct Edge{
    int u;
    int v;
    int w;
};
Edge edge[maxm];
int n,m,k,cnt=0,ans=0;
int f[maxn];

inline int find(int k){
    return f[k]==k?k:f[k]=find(f[k]);
}

void unionn(int x,int y){
    x=find(x);
    y=find(y);
    f[x]=y;
}

bool cmp(Edge a,Edge b){
    return a.w<b.w?1:0;
}

void kruskal(){
    for(int i=1;i<=m;i++){
        register int x,y;
        x=find(edge[i].u);
        y=find(edge[i].v);
        if(x!=y){
            ans+=edge[i].w;//求最小花费
            cnt++;
            unionn(edge[i].u,edge[i].v);
        }
        if(cnt==n-k)
            break;
    }
}

int main(){
    scanf("%d%d%d",&n,&m,&k);
    if(m<n-k){//边数显然不能少于n-k
        printf("No Answer");
        return 0;
    }
    for(int i=1;i<=n;i++)
        f[i]=i;//认自己为父亲
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
    sort(edge+1,edge+m+1,cmp);//克鲁斯卡尔不可缺少的排序
    kruskal();
    printf("%d",ans);
    return 0;
}

友情链接：安利一只小姐姐的博客