最小生成树口袋的天空

Posted 2020-09-28 wxjor

口袋的天空

背景

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

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

描述

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

样例输入1

3 1 2
1 2 1

Copy

样例输出1

1

Copy

限制

每个测试点1s

提示

样例2：
Input:
3 1 1
1 2 1

Output:
No Answer

来源

lolanv

 

试题分析

    最小生成树模板题，直接将最后一个棉花糖连成最小生成树，剩下K-1个用单个的云朵来当棉花糖就好了

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
inline int read(){
	int x=0,f=1;char c=getchar();
	for(;!isdigit(c);c=getchar()) if(c==‘-‘) f=-1;
	for(;isdigit(c);c=getchar()) x=x*10+c-‘0‘;
	return x*f;
}
const int INF = 9999999;
struct data{
	int x,y,v;
}e[1000001];
bool cmp(data a,data b){
	return a.v<b.v;
}
int fa[100001];
int find(int x){
	if(x!=fa[x]) return fa[x]=find(fa[x]);
	return x;
}
int N,M,K; int ans;
int main(){
	N=read(),M=read(),K=read();
	for(int i=1;i<=M;i++){
	    e[i].x=read(),e[i].y=read();
	    e[i].v=read();
	}
	sort(e+1,e+M+1,cmp); int cnt=N;
	for(int i=1;i<=N;i++) fa[i]=i;
	for(int i=1;i<=M&&cnt>K;i++){
		int u=find(e[i].x),v=find(e[i].y);
		if(u!=v){
			ans+=e[i].v;
			cnt--;fa[v]=u;
		}
	}
	printf("%d\n",ans);
	return 0;
}