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

很显然的发现,这是一道DSU……

于是每次合并的时候把权值算上

最后开一个计数器统计DSU的个数就行了,个数一开始是N个,每合并两个集合就少一个。

于是,

#include<cstdio>
#include<iostream> 
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm> 
#include<iterator>
using namespace std;

struct ed
{
	int from,to,w;
}edge[10010]={0};
int fx,tx,wx,edgenum,nodenum,k,ufsnum,ans=0;
int fatherx[1010]={0},ufsc[1010]={0};

bool cmp(ed a,ed b) 
{
	return a.w<b.w;
}

int findx(int e)
{
	if (fatherx[e]!=e) fatherx[e]=findx(fatherx[e]);
	return fatherx[e];
}

void mergex(int e1,int e2,int cost)
{
	fatherx[findx(e1)]=fatherx[findx(e2)];
	ufsc[fatherx[findx(e2)]]+=cost;
}

int main()
{
	scanf("%d%d%d",&nodenum,&edgenum,&k);  
	ufsnum=nodenum; 
	for (int i=1;i<=nodenum;i++) fatherx[i]=i;
	for (int i=1;i<=edgenum;i++) scanf("%d%d%d",&edge[i].from,&edge[i].to,&edge[i].w);
	
	sort(edge+1,edge+1+edgenum,cmp); 
	
	for (int i=1;i<=edgenum;i++)
	{
		if (findx(edge[i].from)!=findx(edge[i].to))
		{
			mergex(edge[i].from,edge[i].to,edge[i].w);
			ufsnum--;
//			cout<<fatherx[edge[i].from]<<" "<<fatherx[edge[i].to]<<endl;
		}
		if (ufsnum==k) break;
	}
	
	if (k>ufsnum)
	{
		printf("No Answer\n");
		return 0;
	}
	
	for (int i=1;i<=nodenum;i++) ans+=ufsc[i];
	printf("%d\n",ans);
	
	return 0;
}

 


发表评论

电子邮件地址不会被公开。 必填项已用*标注

This site uses Akismet to reduce spam. Learn how your comment data is processed.