POJ 2349 Arctic Network

题目大意：

        现有T个测例，每个测例中有N个岗哨和S个卫星频道（1 ≤ S ≤ 100，S < N ≤ 500），接下来给出N个点的坐标(Xi, Yi)，其中0 ≤ Xi, Yi ≤ 10,000，拥有卫星频道的任意两个岗哨可以自由通信不受距离限制，由于卫星频道有限，因此没分配到卫星频道的岗哨之间通信必须通过无线电收发机，可以间接通信，比如A通过B和C通信，其中AB、BC的通信方式可以任意，但是无线电通信是受距离限制的，任意两点之间无线电通信距离不得超过D（D是浮点数），对于每个测例要求你设计一种最优的分配方案使得D最小，只要求输出最小的D，保留两位小数。

题目链接

注释代码：

/*                                  
 * Problem ID : POJ 2349 Arctic Network
 * Author     : Lirx.t.Una                                  
 * Language   : C++                      
 * Run Time   : 16 ms                                  
 * Run Memory : 1456 KB                                  
*/ 

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>

//结论：答案即为最小生成树的第S条边，重要结论！！最小生成树的第K大边的值唯一，不论最小生成树有多少种！！

//距离平方的无穷大（大于10000^2 × 2）
#define	INF		2000000000
//最大点数
#define	MAXV	500

using namespace std;

struct	Point {

	int		x, y;

	int
	POW(int a) { return a * a; }

	int
	operator^(Point &oth) { 
		
		return POW( x - oth.x ) + POW( y - oth.y ); 
	}
};

struct	Node {//Prim堆优化中的结点

	int		u;//结点所代表的点
	int		dp;//结点到当前未完成的生成树的当前最短距离

	Node(void) {}

	Node(int uu, int dd) : u(uu), dp(dd) {}

	bool//构造dp的小顶堆
	operator<(const Node &oth)
	const {
	
		return dp > oth.dp;
	}
};

Point	p[MAXV + 1];//点

int		w[MAXV + 1][MAXV + 1];//weight，计算任意两点之间的欧几里得距离的平方
int		ans[MAXV + 1];//存储最后最小生成树中的所有边
int		dist[MAXV + 1];
bool	vist[MAXV + 1];

double
prim( int s, int n ) {

	int		i;
	int		e;//当前构造出的生成树的边数
	int		u, v;

	Node	node;

	priority_queue<Node>	heap;

	memset(vist, 0, sizeof(vist));

	vist[1] = true;
	for ( i = 2; i <= n; i++ ) {

		dist[i] = w[1][i];
		heap.push(Node( i, dist[i] ));
	}

	e = 1;
	while (true) {//任意两点之间都有边所以必定有解

		while (true) {//必定有解
		
			node = heap.top();
			u	 = node.u;

			heap.pop();

			if ( !vist[u] ) {
			
				vist[u] = true;
				ans[e++] = node.dp;

				break;
			}
		}

		if ( e == n ) break;//构造完毕

		for ( v = 2; v <= n; v++ )
			if ( !vist[v] && w[u][v] < dist[v] ) {
			
				dist[v] = w[u][v];
				heap.push(Node( v, dist[v] ));//更优的结点入堆
			}
	}

	sort(ans, ans + e);//从小到大排
	return sqrt((double)ans[e - s]);//第e - s + 1为第S大边
}

int
main() {

	int		t;//测例数
	int		s, n;//卫星数，岗哨数

	int		i, j;

	scanf("%d", &t);
	while ( t-- ) {

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

		for ( i = 1; i <= n; i++ )
			for ( j = i + 1; j <= n; j++ )
				w[i][j] = w[j][i] = p[i] ^ p[j];

		printf("%.2lf\n", prim( s, n ));
	}

	return 0;
}

无注释代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>

#define	INF		2000000000
#define	MAXV	500

using namespace std;

struct	Point {

	int		x, y;

	int
	POW(int a) { return a * a; }

	int
	operator^(Point &oth) { 
		
		return POW( x - oth.x ) + POW( y - oth.y ); 
	}
};

struct	Node {

	int		u;
	int		dp;

	Node(void) {}

	Node(int uu, int dd) : u(uu), dp(dd) {}

	bool
	operator<(const Node &oth)
	const {
	
		return dp > oth.dp;
	}
};

Point	p[MAXV + 1];

int		w[MAXV + 1][MAXV + 1];
int		ans[MAXV + 1];
int		dist[MAXV + 1];
bool	vist[MAXV + 1];

double
prim( int s, int n ) {

	int		i;
	int		e;
	int		u, v;

	Node	node;

	priority_queue<Node>	heap;

	memset(vist, 0, sizeof(vist));

	vist[1] = true;
	for ( i = 2; i <= n; i++ ) {

		dist[i] = w[1][i];
		heap.push(Node( i, dist[i] ));
	}

	e = 1;
	while (true) {

		while (true) {
		
			node = heap.top();
			u	 = node.u;

			heap.pop();

			if ( !vist[u] ) {
			
				vist[u] = true;
				ans[e++] = node.dp;

				break;
			}
		}

		if ( e == n ) break;

		for ( v = 2; v <= n; v++ )
			if ( !vist[v] && w[u][v] < dist[v] ) {
			
				dist[v] = w[u][v];
				heap.push(Node( v, dist[v] ));
			}
	}

	sort(ans, ans + e);
	return sqrt((double)ans[e - s]);
}

int
main() {

	int		t;
	int		s, n;

	int		i, j;

	scanf("%d", &t);
	while ( t-- ) {

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

		for ( i = 1; i <= n; i++ )
			for ( j = i + 1; j <= n; j++ )
				w[i][j] = w[j][i] = p[i] ^ p[j];

		printf("%.2lf\n", prim( s, n ));
	}

	return 0;
}