Vijos P1234 口袋的天空

口袋的天空

链接

给出一些相连的云，求连接出k棵树所需的最小代价，每棵树最少有一朵云

最小生成树  可以用Kruskal做

思路

一棵树中如果只有一个节点则代价为0

那么可以让尽量多的树中只有一个节点

还有就是考虑无解的情况

当要求的生成树个数大于总节点数时无法构造解

相等时解为0

如果要构造出k棵树的话 考虑在k-1棵树中只放一个节点

那么剩下一棵树中就需要放(n-(k-1))即(n-k+1)个节点

生成树的话就要连n-k条边 如果所给变数小于n-k 无解

问题转化为在m条边中选择n-k条边生成树 搞Kruskal即可

代码

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int Maxn=10001;

struct node {int u,v,c;}line[Maxn];
int n,m,k,ans,tot,fa[Maxn];

bool comp(node a,node b){return a.c<b.c;}

int find(int x)
{
	if(fa[x]==x)return x;
	int r=find(fa[x]);
	fa[x]=r;
	return r;
}

int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++)
		scanf("%d%d%d",&line[i].u,&line[i].v,&line[i].c);
	
	if(k>n){
		printf("No Answer");
		return 0;
	}
	else if(k==n){
		printf("0");
		return 0;
	}
	
	k=n-k;
	
	if(k>m){
		printf("No Answer");
		return 0;
	}
	
	sort(line+1,line+m+1,comp);
	
	for(int i=1;i<=m;i++){
		int r1=find(line[i].u);
		int r2=find(line[i].v);
		if(r1!=r2){
			ans+=line[i].c;
			fa[r1]=r2;
			tot++;
		}
		if(tot==k)break;
	}
	
	printf("%d",ans);
	
	return 0;
}

欢迎指出Bug

End。