BZOJ 2626: JZPFAR KDtree + 堆
escription
平面上有n个点。现在有m次询问,每次给定一个点(px, py)和一个整数k,输出n个点中离(px, py)的距离第k大的点的标号。如果有两个(或多个)点距离(px, py)相同,那么认为标号较小的点距离较大。
Input
第一行,一个整数n,表示点的个数。
下面n行,每行两个整数x_i, y_i,表示n个点的坐标。点的标号按照输入顺序,分别为1..n。
下面一行,一个整数m,表示询问个数。
下面m行,每行三个整数px_i, py_i, k_i,表示一个询问。
下面n行,每行两个整数x_i, y_i,表示n个点的坐标。点的标号按照输入顺序,分别为1..n。
下面一行,一个整数m,表示询问个数。
下面m行,每行三个整数px_i, py_i, k_i,表示一个询问。
Output
m行,每行一个整数,表示相应的询问的答案。
题解: 和之前一道题几乎一模一样,直接用堆 + $KDtree$ 维护即可.
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 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 | #include<bits/stdc++.h> #define maxn 200000 #define inf 1000000000000000 #define mid ((l+r)>>1) #define rson(x) (t[x].ch[1]) #define lson(x) (t[x].ch[0]) #define ll long long using namespace std; void setIO(string s) { string in = s + ".in" ; freopen (in.c_str(), "r" , stdin); } int d, n, m; struct Node { ll dis; int k; Node (ll dis = 0, int k = 0) : dis(dis), k(k) {} bool operator < (Node b) const { return dis == b.dis ? b.k > k : b.dis < dis; } }; priority_queue <Node> Q; // return if a is less than b bool cmp1(Node a, Node b) { return a.dis == b.dis ? a.k > b.k : a.dis < b.dis; } ll sqr(ll a) { return a * a; } struct ND { int ch[2], id; ll p[2], minv[2], maxv[2]; }t[maxn],T; bool cmp(ND a, ND b) { return a.p[d] == b.p[d] ? a.p[d ^ 1] < b.p[d ^ 1] : a.p[d] < b.p[d]; } void pushup( int x, int y) { for ( int i = 0; i < 2 ; ++i) { t[x].minv[i] = min(t[x].minv[i], t[y].minv[i]); t[x].maxv[i] = max(t[x].maxv[i], t[y].maxv[i]); } } int build( int l, int r, int o) { d = o; nth_element(t + l, t + mid, t + 1 + r, cmp); for ( int i = 0; i < 2 ; ++i) t[mid].minv[i] = t[mid].maxv[i] = t[mid].p[i]; lson(mid) = rson(mid) = 0; if (mid > l) { lson(mid) = build(l, mid - 1, o ^ 1); pushup(mid, lson(mid)); } if (r > mid) { rson(mid) = build(mid + 1, r, o ^ 1); pushup(mid, rson(mid)); } return mid; } ll getmax( int x) { ll ans = 0; for ( int i = 0; i < 2 ; ++i) { ans += max(sqr(t[x].minv[i] - T.p[i]), sqr(t[x].maxv[i] - T.p[i])); } return ans; } void query( int x, ll x1, ll y1) { ll cur = sqr(t[x].p[0] - x1) + sqr(t[x].p[1] - y1), dl, dr, dn = getmax(x); if (dn < Q.top().dis) return ; if (cmp1(Q.top(), Node(cur, t[x].id))) { Q.pop(); Q.push(Node(cur, t[x].id)); } dl = lson(x) ? getmax(lson(x)) : -inf; dr = rson(x) ? getmax(rson(x)) : -inf; if (dl > dr) { if (dl >= Q.top().dis) query(lson(x), x1, y1); if (dr >= Q.top().dis) query(rson(x), x1, y1); } else { if (dr >= Q.top().dis) query(rson(x), x1, y1); if (dl >= Q.top().dis) query(lson(x), x1, y1); } } int main() { int i, j, k, root; ll x, y; // setIO("input"); scanf ( "%d" ,&n); for (i = 1; i <= n ; ++i) scanf ( "%lld%lld" ,&t[i].p[0],&t[i].p[1]), t[i].id = i; root = build(1, n, 0); scanf ( "%d" ,&m); while (m --) { scanf ( "%lld%lld%d" ,&x,&y,&k); while (!Q.empty()) Q.pop(); for (i = 1; i <= k ; ++i) Q.push(Node(-2333, -1)); T.p[0] = x, T.p[1] = y; query(root, x, y); printf ( "%d\n" ,Q.top().k); } return 0; } |
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 开发者新选择:用DeepSeek实现Cursor级智能编程的免费方案
· Tinyfox 发生重大改版
· 独立开发经验谈:如何通过 Docker 让潜在客户快速体验你的系统
· 小米CR6606,CR6608,CR6609 启用SSH和刷入OpenWRT 23.05.5
· 近期最值得关注的AI技术报告与Agent综述!