2349 Arctic Network prim最小生成树 基础

图我们构造完成，调用prim算法，把该算法中每一次贪心得到的最小权边存入一个数组中。算法结束后对那个数组sort递减排序。这样，前边最大的s个用那卫星让他们通信去。然后从第s开始配给他们足够功率的对讲机使。

#include<iostream>
using namespace std;
#include<algorithm>
#include<math.h>
#define max 505
double map[max][max];
double dis[max];
int s,p;
int ta[max],tb[max];
bool vis[max];
double sum[max];
double ca(int x,int y,int a,int b)
{
double t=(x-a)*(x-a)+(y-b)*(y-b);
return sqrt(t);
}
bool cmp(double x,double y)
{
return x>y;
}
int t=1;
void prim()
{
for(int i=1;i<=p;i++)
{
int temp;
double cf=9999999.0;
for(int j=1;j<=p;j++)
{
if(!vis[j]&&dis[j]<cf)
{
temp=j;
cf=dis[j];
}
}
vis[temp]=true;
if(cf==9999999.0)
cf=0;
sum[t++]=cf;
for(int j=1;j<=p;j++)
{
if(!vis[j]&&dis[j]>map[temp][j])
dis[j]=map[temp][j];
}
}
return ;
}

int main()
{
int n;
cin>>n;
while(n--)
{
t=1;
memset(vis,false,sizeof(vis));
memset(map,9999999.0,sizeof(map));

scanf("%d%d",&s,&p);
for(int i=1;i<=p;i++)
scanf("%d%d",&ta[i],&tb[i]);
for(int i=1;i<=p;i++)
{
for(int j=1;j<=p;j++)
if(j!=i)
map[i][j]=map[j][i]=ca(ta[i],tb[i],ta[j],tb[j]);
}
vis[1]=true;
for(int i=1;i<=p;i++)
dis[i]=map[1][i];
prim();
printf("%.2lf\n",sum[s]);
}
return 0;
}