poj_2349 Kruskal 最小生成树
题目大意
给定N个点的坐标，这N个点之间需要进行通讯。通讯方式可以采用卫星通信或无线通信，若两点之间采用为卫星通信，则两点之间的距离无限制，若采用无线通讯，则两点之间的距离不能大于某个值D。
现有s台卫星通信设备可以分配给这N个点，其余的点之间必须使用无线通信。要让这N个点中所有的点都能相互通信，则合理分配s台卫星通信设备，可以使得采用无线通信的那些点之间的距离D达到一个最小值，求该最小值。
题目分析
让所有的点之间均能通信，为一个生成树结构。题目就是求出这N个点的最小生成树。然后将最小生成树分成S个割集，割集之间采用卫星通信，割集之内采用无线通信，求出割集之内点之间的最大值即可。
可以证明，割集之内的点距离的最大值的最小值为最小生成树的N-1条边从大到小排列后第S个边长。采用kruskal算法解决。
实现(c++)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 | #include<stdio.h>#include<queue>#include<vector>#include<cmath>using namespace std;#define MAX_NODE 505//点的数据结构struct Point{ int x; int y;};vector<Point> gPoints;//边的数据结构struct Edge{ int from; int to; double dist; Edge(int f, int t, double d) : from(f), to(t), dist(d){};};vector<Edge> gEdges;//计算两点之间的距离double Dist(const Point& p1, const Point& p2){ return sqrt(1.0*(p1.x - p2.x)*(p1.x - p2.x) + (p1.y - p2.y)*(p1.y - p2.y));}//用并查集来判断加入一条边是否会构成环int gRoot[MAX_NODE];int GetRoot(int c){ if (gRoot[c] != c){ gRoot[c] = GetRoot(gRoot[c]); } return gRoot[c];}bool SameRoot(int c1, int c2){ int p1 = GetRoot(c1); int p2 = GetRoot(c2); return p1 == p2;}void Union(int c1, int c2){ int p1 = GetRoot(c1); int p2 = GetRoot(c2); gRoot[p1] = p2;}//用于对边进行排序bool Compare(const Edge& e1, const Edge& e2){ return e1.dist < e2.dist;}double Kruskal(int s, int n){ double result; for (int i = 0; i < n; i++){ gRoot[i] = i; } sort(gEdges.begin(), gEdges.end(), Compare); //无向图的边只存储了 从序号较小的节点指向序号较大的节点 int count = 0; for (int i = 0; i < gEdges.size(); i++){ Edge& e = gEdges[i]; if (SameRoot(e.from, e.to)) continue; count++; if (count == n - s){ //从最小生成树中的n-1条边，去掉最大的s-1条边（因为有s个卫星站，相当于s个点，则s-1条边） //，剩下的n-1-s条边中，最大的边长即为所求 result = e.dist; return result; } Union(e.to, e.from); //gRoot[gRoot[e.to]] = gRoot[e.from]; //注意合并的时候，将 to 的根更新为 from的根。因为所有的边只存储了从小序号指向大序号 } return 0;}int main(){ int cas, s, p; Point point; scanf("%d", &cas); while (cas--){ scanf("%d %d", &s, &p); gEdges.clear(); gPoints.clear(); for (int i = 0; i < p; i++){ scanf("%d %d", &point.x, &point.y); for (int j = 0; j < i; j++){ double dist = Dist(point, gPoints[j]); gEdges.push_back(Edge(j, i, dist)); } gPoints.push_back(point); } double result = Kruskal(s, p); printf("%.2lf\n", result); } return 0;} |
