POJ 2349 Arctic Network 最小生成树题解

本题也是使用Prime和Kruskal都可以的最小生成树的题解。

本题一点新意就是：需要除去最大的S-1个距离，因为可以使用卫星覆盖这些距离。

技巧：建图建有向图，速度快点，不用计算两边。

这里使用Prime，因为是稠密图。

#include <stdio.h>#include <math.h>#include <stdlib.h>#include <string.h>#include <limits.h>#include <algorithm>using std::sort;const int MAX_VEC = 501;struct Pos{int x, y;};float gra[MAX_VEC][MAX_VEC];Pos pos[MAX_VEC];float dist[MAX_VEC];bool vis[MAX_VEC];int N, S, P;float Prime(){memset(vis, 0, sizeof(bool) * P);vis[0] = true;for (int i = 0; i < P-1; i++){float m = (float)INT_MAX;int id = 0;for (int j = 0; j < P; j++){if (!vis[j] && gra[0][j] < m){m = gra[0][j];id = j;}}vis[id] = true;dist[i] = m;for (int j = 0; j < id; j++){if (!vis[j] && gra[0][j] > gra[j][id]) gra[0][j] = gra[j][id];;}for (int j = id+1; j < P; j++){if (!vis[j] && gra[0][j] > gra[id][j]) gra[0][j] = gra[id][j];}}sort(dist, dist+P-1);return dist[P-1-S];}int main(){scanf("%d", &N);while (N--){scanf("%d %d", &S, &P);for (int i = 0; i < P; i++){scanf("%d %d", &pos[i].x, &pos[i].y);}for (int i = 0; i < P; i++){for (int j = i+1; j < P; j++){float a = float(pos[i].x - pos[j].x);float b = float(pos[i].y - pos[j].y);gra[i][j] = sqrtf(a*a + b*b);}}printf("%.2f\n", Prime());}return 0;}