ZOJ 1914 Arctic Network （POJ 2349） MST

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1914

http://poj.org/problem?id=2349

题目大意，给定一些点的坐标，求MST，然后要求求去掉最大的k条边后，最大的边

直接Prim，然后在排序即可。

小技巧是一开始不求平方根，最后输出的时候在求出平方根即可。

ZOJ上排行第三，不过在POJ就被虐了。。。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAXN=501;
const int INF=9999999;
int map[MAXN][MAXN];
double dis[MAXN];
struct point
{
int x,y;
}ship[MAXN];

void prim(int n)
{
int i,j;
for(i=1;i<=n;i++)
dis[i]=INF;

bool vis[MAXN]={0};

int cur=1;
vis[cur]=1;
dis[cur]=0;

for(i=1;i<=n;i++)
{
double mini=INF;
for(j=1;j<=n;j++)
if(!vis[j] && dis[j] > map[cur][j])
dis[j]=map[cur][j];

for(j=1;j<=n;j++)
if(!vis[j] && mini > dis[j])
mini=dis[cur=j];

vis[cur]=true;
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,x;
scanf("%d%d",&x,&n);

int i,j;
for(i=1;i<=n;i++)
scanf("%d%d",&ship[i].x,&ship[i].y);

for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
map[i][j]=	map[j][i] =
(ship[j].y -ship[i].y) *(ship[j].y -ship[i].y) + (ship[j].x -ship[i].x)*(ship[j].x -ship[i].x);
}
}

prim(n);

sort(dis+1,dis+n+1);
printf("%.2lf\n",sqrt(dis[n-x+1]));
}
return 0;
}