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;
}