洛谷P1195 口袋的天空

题目背景

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

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

题目描述

给你云朵的个数 𝑁N，再给你 𝑀M 个关系，表示哪些云朵可以连在一起。

现在小杉要把所有云朵连成 𝐾K 个棉花糖，一个棉花糖最少要用掉一朵云，小杉想知道他怎么连，花费的代价最小。

输入格式

第一行有三个数N,M,K。

接下来 M 行每行三个数X,Y,L，表示 X 云和 Y 云可以通过 L 的代价连在一起。

输出格式

对每组数据输出一行，仅有一个整数，表示最小的代价。

如果怎么连都连不出 K 个棉花糖，请输出 No Answer。

输入输出样例

输入 #1复制

3 1 2
1 2 1

输出 #1复制

1

说明/提示

对于 30% 的数据，1≤N≤100，1≤M≤10^3；

对于 100% 的数据，1≤N≤10^3，1≤M≤10^4，1≤K≤10，1≤X,Y≤N，0≤L<10^4。

代码： 

#include <bits/stdc++.h>
using namespace std;
int n,m,k,he,f;
pair<int,pair<int,int>> sn[1000010];
int anc[1000010];
int sum,cnt;
int find(int x){//并查集模板 
	if(x!=anc[x]) anc[x]=find(anc[x]);
	return anc[x];
}
void kruskal(){//模板 小改 
	he=n;
	for(int i = 1;i<=n;i++) anc[i]=i;
	for(int i = 1;i<=m;i++){
		int a=sn[i].second.first,b=sn[i].second.second,c=sn[i].first;
		int t1=find(a),t2=find(b);
		if(t1!=t2){
			anc[t1]=t2;
			sum+=c;
			//每合并一次就少一个集合，即少一个棉花糖 
			he--;
		}
		if(he==k){
		//!!!注意是当合并到只剩k个集合时，停止合并，而不是算合并了多少条边 
			break;
		}
	}
	if(he==k) cout << sum;//如果不能合并成k个，说明不成功 
	else cout << "No Answer";
}
int main(){
	cin >> n >> m >> k;
	for(int i = 1;i<=m;i++){
		int x,y,l;
		cin >> x >> y >> l;
		sn[i]={l,{x,y}};
	}
	sort(sn+1,sn+m+1);
	kruskal();
	return 0;
}