【P1195 口袋的天空】

口袋的天空

题目背景

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

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

题目描述

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

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

输入格式

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

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

输出格式

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

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

样例 #1

样例输入 #1

3 1 2
1 2 1

样例输出 #1

1

提示

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

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

题意及分析

给出n个点，m个关系，把这些点连成k棵树，每棵树至少由一个点组成，问怎么连使总的代价最小。
很明显的最小生成树问题，经典的最小生成树问题是：给n个点，m条边，把n个点连成一棵树，问怎么连是总的代价最小，当然最小生成树也可能不存在。
而这道题又是恰好是Kruskal算法的变式。
Kruskal算法流程：
将m条边按权值从小到大排序后，枚举每一条边，如果该边不在连通块中，则加入连通块，直到选了n-1条边为止。
我们知道一棵n个点的树有n-1条边，那么在这道题中，我们把一棵树分成k棵树就需要去掉k-1条边，即需要连：n-1-（k-1) = n-k条边。

放代码

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e4+5;
const int INF=0x3f3f3f3f;
int n,m,k;
int map[N][N];
int f[N];
bool st[N];

struct edge{
	int u,v,w;
	
	bool operator < (const edge &W) const{
		return w<W.w;
	};
	
}edges[N]; 

int find(int x){
	if(f[x]!=x) f[x]=find(f[x]);
	return f[x];
}

int main(){
	scanf("%d %d %d",&n,&m,&k);
	if(k>n){
		printf("No Answer\n");
		return 0;
	}
	int u,v,w;
	for(int i=0;i<m;++i){
		scanf("%d %d %d",&u,&v,&w);
		edges[i].u=u,edges[i].v=v,edges[i].w=w; 
	}
	sort(edges,edges+m);

	for(int i=1;i<=n;++i) f[i]=i;
	
	int cnt=0,sum=0;
	for(int i=0;i<m;++i){
		u=edges[i].u,v=edges[i].v,w=edges[i].w;
		int f_u=find(u);
		int f_v=find(v);
		if(f_u!=f_v){
			f[f_u]=f_v;
			++cnt;
			sum+=w;
		}
		if(cnt>=n-k){
			printf("%d",sum);
			return 0;
		}
	}
	printf("No Answer");
	return 0;
}