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,表示一个询问。

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;
}

  

posted @   EM-LGH  阅读(184)  评论(0编辑  收藏  举报
编辑推荐:
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
阅读排行:
· 开发者新选择:用DeepSeek实现Cursor级智能编程的免费方案
· Tinyfox 发生重大改版
· 独立开发经验谈:如何通过 Docker 让潜在客户快速体验你的系统
· 小米CR6606,CR6608,CR6609 启用SSH和刷入OpenWRT 23.05.5
· 近期最值得关注的AI技术报告与Agent综述!
点击右上角即可分享
微信分享提示