首页 > 代码库 > POJ 3714 Raid 最近对点题解
POJ 3714 Raid 最近对点题解
本题是一般最近对点求解,稍微增加点限定:有两个集合点,要求不同集合中的点的最近对。
那么就增加一个判断,如果是同一个集合中的点,那么就返回最大值,其他和一般的最近对点解法一样。
注意:本题数据有重合点,那么就要防止分类的时候溢出。
Geeks上的最近对的程序是无法处理有重合点的情况的。
#include <stdio.h> #include <stdlib.h> #include <float.h> #include <math.h> #include <algorithm> #include <string.h> using std::sort; struct Point { int x, y; int whichSet; }; inline int xCmp(const void *a, const void *b) { const Point *a1 = static_cast<const Point *>(a); const Point *b1 = static_cast<const Point *>(b); return a1->x - b1->x; } inline int xCmp2(const Point &a, const Point &b) { return a.x < b.x; } inline int yCmp2(const Point &a, const Point &b) { return a.y < b.y; } inline int yCmp(const void *a, const void *b) { const Point *a1 = static_cast<const Point *> (a); const Point *b1 = static_cast<const Point *> (b); return a1->y - b1->y; } inline float dist(Point &a, Point &b) { if (a.whichSet == b.whichSet) return FLT_MAX; float xd = (float)(a.x - b.x); float yd = (float)(a.y - b.y); return sqrtf(xd*xd + yd*yd); } float bruteForce(Point P[], int n) { float m = FLT_MAX; for (int i = 0; i < n; ++i) { for (int j = i+1; j < n; ++j) { float d = dist(P[i], P[j]); if (d < m) m = d; } } return m; } inline float mMin(float x, float y) { return x < y? x : y; } float stripClosest(Point strip[], int n, float d) { for (int i = 0; i < n; i++) { for (int j = i+1; j < n && strip[j].y -strip[i].y < d; j++) { float t = dist(strip[i], strip[j]); if (t < d) d = t; } } return d; } float closestUtil(Point Px[], Point Py[], int n) { if (n <= 3) return bruteForce(Px, n); int m = n >> 1; Point midPoint = Px[m]; Point *PyL = new Point[m+1]; Point *PyR = new Point[n-m-1]; int le = 0, ri = 0; for (int i = 0; i < n; i++) {//修正bug:增加le<m+1判断,防止重复点,引起溢出 if (Py[i].x <= midPoint.x && le < m+1) PyL[le++] = Py[i]; else PyR[ri++] = Py[i]; } float dl = closestUtil(Px, PyL, le);//m+1); float dr = closestUtil(Px+m+1, PyR, ri);//n-m-1); float d = mMin(dl, dr); Point *strip = new Point[n]; int j = 0; for (int i = 0; i < n; i++) { if (fabsf(float(Py[i].x - midPoint.x)) < d) strip[j++] = Py[i]; } d = mMin(d, stripClosest(strip, j, d)); delete [] strip; delete [] PyL; delete [] PyR; return d; } float closest(Point P[], int n) { Point *Px = new Point[n]; Point *Py = new Point[n]; memcpy(Px, P, n * sizeof(Point)); memcpy(Py, P, n * sizeof(Point)); //qsort(Px, n, sizeof(Point), xCmp); sort(Px, Px+n, xCmp2); //qsort(Py, n, sizeof(Point), yCmp); sort(Py, Py+n, yCmp2); float d = closestUtil(Px, Py, n); delete [] Px; delete [] Py; return d; } int main() { int T, n; scanf("%d", &T); while (T--) { scanf("%d", &n); Point *P = new Point[n<<1]; for (int i = 0; i < n; i++) { scanf("%d %d", &P[i].x, &P[i].y); P[i].whichSet = 1; } for (int i = n; i < (n<<1); i++) { scanf("%d %d", &P[i].x, &P[i].y); P[i].whichSet = 2; } printf("%.3f\n", closest(P, n<<1)); delete [] P; } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。