CodingTrip - 携程编程大赛 （预赛第一场）第三题 携程全球数据中心建设

//球面上2点的距离计算 + 最小生成树

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

const double pi = 3.14159265358979323846;
double d,r,sum,cur,n;

struct node {
double j, w;
node(){};
node(double _x, double _y): j(_x), w(_y) {}
};
node tt[105];

inline double distence(double j1, double w1, double j2, double w2) {
return r*acos(cos(w1)*cos(w2)*cos(j2-j1) + sin(w1)*sin(w2));
}

double map[105][105];
bool visited[105];
double dist[105];

int MinVertex()
{
int k = -1;
double minValue=pi*d;

for (int i=0;  i<n; i++)
{
if (dist[i] < minValue && visited[i]==false)
{
minValue=dist[i];
k = i;
}
}
return k;
}

double prim()
{
for(int i=0;i<n;i++) dist[i]=map[0][i];
fill(visited,visited+105,false);

visited[0]=true;
double cost=0;

while(true)
{
int K=MinVertex();

if (K==-1) break;
visited[K]=true;
cost+=dist[K];

for (int i=0;i<n;i++)
if (visited[i]==false && map[K][i]<dist[i]) dist[i]=map[K][i];
}

return cost;
}

void run()
{
cin>>d>>cur>>n;
r=d/2.0;

double j, w;
for(int i=0;i<n;i++) {
scanf("%lf %lf", &w, &j);
w = w*pi/180.0;
j = j*pi/180.0;
tt[i] = node(j, w);
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++)
{
double dis;
if(i==j) dis=0;
else dis = distence(tt[i].j, tt[i].w, tt[j].j, tt[j].w);
map[i][j]=map[j][i]=dis;
}
}
sum=prim();
if(sum>cur) printf("N\n");
else printf("Y\n");
}
int main() {
int T;
cin>>T;
while(T--) run();
return 0;
}