题目链接:
http://poj.org/problem?id=2349
题意分析:
有P个哨所和S个卫星通道,任意两个哨所的连接可以为(1)用一个卫星通道相连(2)两个哨所的距离小于等于D,求满足任意两个哨所直接或间接相连D的最小值。
解题思路:
题目要求最后任意哨所两两相连,这就转化成最小生成树的问题。
现假设我们已经求得满足要求的最小生成树,以贪心的思维可以想到,每一个卫星通道都应该使用在这个最小生成树里权值最大的边。
由此,题目便转化成求图的最小生成树中第(n-1-s)大的边。用kruskal算法求解。
AC代码(POJ提交时language选择C++):
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN = 550;
const int MAXM = 250050;
int T,s,n;
int tol;
double len[MAXN];
int F[MAXN];//并查集使用
struct point
{
double x,y;
}p[MAXN];
struct Edge
{
int u,v;
double w;
bool operator < (const Edge & e)const
{
return w<e.w;
}
}edge[MAXM];
void addedge(int u,int v,double w)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol++].w=w;
}
int find(int x)
{
if(F[x]==-1)return x;
else return F[x]=find(F[x]);
}
void kruskal()
{
memset(F,-1,sizeof(F));
sort(edge,edge+tol);
int cnt=0;
for(int i=0;i<tol;i++)
{
int u=edge[i].u;
int v=edge[i].v;
double w=edge[i].w;
int t1=find(u);
int t2=find(v);
if(t1!=t2)
{
len[cnt]=w;
F[t1]=t2;
cnt++;
}
if(cnt==n-1)break;
}
printf("%.2lf\n",len[n-1-s]);
}
double dis(int a,int b)
{
return sqrt((p[a].x-p[b].x)*(p[a].x-p[b].x)+(p[a].y-p[b].y)*(p[a].y-p[b].y));
}
int main()
{
scanf("%d",&T);
while(T--)
{
tol=0;
scanf("%d%d",&s,&n);
for(int i=0;i<n;i++)
scanf("%lf%lf",&p[i].x,&p[i].y);
for(int i=0;i<n-1;i++)
{
for(int j=i+1;j<n;j++)
{
double distance = dis(i,j);
addedge(i,j,distance);
}
}
kruskal();
}
return 0;
}