洛谷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

最小生成树，若有k个棉花糖，则一共要连n-k条边。

由此思路可得代码如下：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 1010
#define MAXM 10010
using namespace std;
int n,m,k,fa[MAXN];
struct node{
	int u,v,w;
}a[MAXM];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
bool cmp(const node &x,const node &y){
	return x.w<y.w;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void uniun(int x,int y){x=find(x);y=find(y);if(x!=y)fa[y]=x;}
void kruskal(){
	int ans=0,s=0;
	for(int i=1;i<=m;i++){
		if(find(a[i].u)!=find(a[i].v)){
			uniun(a[i].u,a[i].v);
			ans+=a[i].w;
			s++;
		}
		if(s==n-k)break;
	}
	printf("%d\n",ans);
}
int main(){
	n=read();m=read();k=read();
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++){a[i].u=read();a[i].v=read();a[i].w=read();}
	sort(a+1,a+m+1,cmp);
	kruskal();
	return 0;
}