最小生成树 套餐

poj 2349 

题意是指去掉大于等于d的边后会形成小于等于s个连通块，问d的最小值。应用最小生成树的一个性质，如果去掉大于等于d的边，最小生成树分为多少个部分，原图也会分成多少个部分，所以取最小生成树的第s条大边即可。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
using namespace std;
struct node{
	double len;
	int from,to;
}tree[251111];
bool cmp(node x,node y)
{
	return x.len<y.len;
}
double dis[555][555],x[555],y[555],path[555];
int fa[555];
int t,n,cnt,s;
int getfa(int x)
{
	if(x!=fa[x]) fa[x]=getfa(fa[x]);
	return fa[x];
}
void solve()
{
	for(int i=1;i<=n;i++)
		fa[i]=i;
	int js=0,flag;
	double sum=0;
	for(int i=1;i<=cnt;i++)
	{
		int a,b;
		a=getfa(tree[i].from);
		b=getfa(tree[i].to);
		if(a!=b)
			{
				fa[a]=b;
				js++;
				path[js]=tree[i].len;
			}
		if(js==n-1)
			{
				flag=1;
				break;
			}
	}
	if(flag==1)
	{
		sort(path+1,path+n);
		if(s==0||s==1)
			printf("%.2f\n",path[n-1]);
		else
			printf("%.2f\n",path[n-s]);
	}
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&s,&n);
		for(int i=1;i<=n;i++)
			scanf("%lf%lf",&x[i],&y[i]);
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				{
					if(j!=i)
						dis[i][j]=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
					else	
						dis[i][j]=0;
				}
		cnt=0;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=i-1;j++)
			{
					tree[++cnt].len=dis[i][j],tree[cnt].from=i,tree[cnt].to=j;
			}
		sort(tree+1,tree+cnt+1,cmp);
		/*for(int i=1;i<=cnt;i++)
			printf("%d %d %f\n",tree[i].from,tree[i].to,tree[i].len);*/
		solve();
	}
}