HDU4347 The Closest M Points
来源:互联网 发布:asp.net 微信支付源码 编辑:程序博客网 时间:2024/05/21 19:03
n维空间最近m点的问题,应该算是经典模型了。
n维:
采用KD树解决,轮流沿着各个维度把点分成两部分(还有一种思路是把跨度最大的那一维度分成两部分,这一种方法是更优的,但是实现起来稍微复杂一点点。而均摊下来两种方法的复杂度相同,所以这里只采用了轮流的建树法),建树即可。
m点:
最近点和最近m点的问题是一样的。
#include <cstdio>#include <cstring>#include <cmath>#include <iostream>#include <vector>#include <queue>#include <stack>#include <algorithm>using namespace std;typedef long long ll;const int maxn = 50010;int n, k, idx, q, m;struct Point{ int x[5]; bool operator < (const Point &o) const {return x[idx] < o.x[idx];}}p[maxn], a, ans[15];typedef pair<double, Point> tp;priority_queue<tp> nq;Point pt[maxn << 2];int son[maxn << 2];void buildKD(int l, int r, int rt = 1, int dep = 0){ if (l > r) return; son[rt] = r - 1; son[rt<<1] = son[rt<<1|1] = -1; idx = dep % k; int mid = (l + r) >> 1; nth_element(p+l, p+mid, p+r+1); pt[rt] = p[mid]; buildKD(l, mid-1, rt<<1, dep+1); buildKD(mid+1, r, rt<<1|1, dep+1);}void find(Point a, int m, int rt = 1, int dep = 0){ if (son[rt] == -1) return; tp node(0, pt[rt]); for (int i = 0; i < k; i++) node.first += ((double)a.x[i]-pt[rt].x[i]) * ((double)a.x[i]-pt[rt].x[i]); int dim = dep % k, lhs = rt << 1, rhs = rt << 1 | 1, flag = 0; if (a.x[dim] >= pt[rt].x[dim]) swap(lhs, rhs); if (son[lhs] != -1) find(a, m, lhs, dep+1); if (nq.size() < m) nq.push(node), flag = 1; else { if (node.first < nq.top().first) nq.pop(), nq.push(node); if ((a.x[dim]-pt[rt].x[dim])*(a.x[dim]-pt[rt].x[dim]) < nq.top().first) flag = 1; } if (son[rhs] != -1 && flag) find(a, m, rhs, dep+1);}int main(){ while (scanf("%d%d", &n, &k) == 2) { for (int i=1;i<=n;i++) for (int j=0;j<k;j++) scanf("%d", &p[i].x[j]); buildKD(1, n); scanf("%d", &q); while (q--) { for (int i=0;i<k;i++) scanf("%d", &a.x[i]); scanf("%d", &m); while (!nq.empty()) nq.pop(); find(a, m); for (int i=0;i<m;i++) ans[i] = nq.top().second, nq.pop(); printf("the closest %d points are:\n", m); for (int i=m-1;i>=0;i--) { for (int j=0;j<k-1;j++) printf("%d ", ans[i].x[j]); printf("%d\n", ans[i].x[k-1]); } } } return 0;}
- HDU4347 The Closest M Points
- hdu4347-The Closest M Points
- HDU4347-The Closest M Points
- HDU4347--The Closest M Points(KD树)
- [KDTree] [优先队列] [HDU4347] The Closest M Points
- hdu4347 The Closest M Points(kdtree+stl)
- The Closest M Points
- [BZOJ3053]The Closest M Points
- poj:4091:The Closest M Points
- 【BZOJ 3053】 The Closest M Points
- HDU 4347 The Closest M Points
- 【35.43%】【hdu 4347】The Closest M Points
- BZOJ3053: The Closest M Points kdtree
- [BZOJ3053][KD树]The Closest M Points
- BZOJ 3053: The Closest M Points kdtree
- hdu 4347 The Closest M Points
- [bzoj3053][kd树]The Closest M Points
- hdu - 4347 - The Closest M Points - kd树
- Android 经验积累之BUG解决方法
- E2 (b)
- python项目目录管理
- Android SDK Manager无法更新的解决
- JavaScript中的匿名函数及函数的闭包
- HDU4347 The Closest M Points
- 【打渔】CodeForce Round 197 Div.2 Problem_A
- 浅谈百度360T盘大战
- Exercises 3.3
- JavaScript跨域总结与解决办法
- CSS3 圆角(border-radius)
- hdu1874畅通工程续
- JS操作iframe
- POJ3256-DFS