[BZOJ2626]JZPFAR（kd-tree+堆）

题目描述

传送门

题解

注意这里是最远的k个点
和BZOJ3053差不多，多了一个限制，就是编号要尽量大
就把维护的堆改一下就行了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
#define LL long long
#define N 100005

struct data
{
    int l,r,id;
    LL dis,d[2],mx[2],mn[2];
    bool operator < (const data &a) const
    {
        return a.dis<dis||a.dis==dis&&a.id>id;
    }
};
int n,m,k,root,cnt;
LL x,y;
data tr[N*4],p,ans;
priority_queue <data> q;

void update(int x)
{
    int l=tr[x].l,r=tr[x].r;
    if (l)
    {
        tr[x].mx[0]=max(tr[x].mx[0],tr[l].mx[0]);
        tr[x].mn[0]=min(tr[x].mn[0],tr[l].mn[0]);
        tr[x].mx[1]=max(tr[x].mx[1],tr[l].mx[1]);
        tr[x].mn[1]=min(tr[x].mn[1],tr[l].mn[1]);
    }
    if (r)
    {
        tr[x].mx[0]=max(tr[x].mx[0],tr[r].mx[0]);
        tr[x].mn[0]=min(tr[x].mn[0],tr[r].mn[0]);
        tr[x].mx[1]=max(tr[x].mx[1],tr[r].mx[1]);
        tr[x].mn[1]=min(tr[x].mn[1],tr[r].mn[1]);
    }
}
int build(int l,int r,int d)
{
    int mid=(l+r)>>1;
    for (int i=l;i<=r;++i) tr[i].dis=tr[i].d[d];
    nth_element(tr+l,tr+mid,tr+r+1);
    tr[mid].mx[0]=tr[mid].mn[0]=tr[mid].d[0];
    tr[mid].mx[1]=tr[mid].mn[1]=tr[mid].d[1];
    if (l<mid) tr[mid].l=build(l,mid-1,d^1);
    if (mid<r) tr[mid].r=build(mid+1,r,d^1);
    update(mid);
    return mid;
}
LL qr(LL x)
{
    return x*x;
}
LL dist(int now)
{
    LL dis=0;
    dis+=max(qr(x-tr[now].mn[0]),qr(x-tr[now].mx[0]));
    dis+=max(qr(y-tr[now].mn[1]),qr(y-tr[now].mx[1]));
    return dis;
}
void query(int now)
{
    LL dl,dr;
    tr[now].dis=qr(tr[now].d[0]-x)+qr(tr[now].d[1]-y);
    if (cnt<k)
    {
        ++cnt;
        q.push(tr[now]);
    }
    else
    {
        data p=q.top();
        if (tr[now].dis>p.dis||tr[now].dis==p.dis&&tr[now].id<p.id)
        {
            q.pop();
            q.push(tr[now]);
        }
    }
    if (tr[now].l) dl=dist(tr[now].l);
    else dl=-1;
    if (tr[now].r) dr=dist(tr[now].r);
    else dr=-1;
    if (dl>dr)
    {
        if (dl!=-1&&(cnt<k||dl>=q.top().dis)) query(tr[now].l);
        if (dr!=-1&&(cnt<k||dr>=q.top().dis)) query(tr[now].r);
    }
    else
    {
        if (dr!=-1&&(cnt<k||dr>=q.top().dis)) query(tr[now].r);
        if (dl!=-1&&(cnt<k||dl>=q.top().dis)) query(tr[now].l);
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i)
    {
        scanf("%lld%lld",&tr[i].d[0],&tr[i].d[1]);
        tr[i].id=i;
    }
    root=build(1,n,0);
    scanf("%d",&m);
    for (int i=1;i<=m;++i)
    {
        scanf("%lld%lld%d",&x,&y,&k);
        cnt=0;
        while (!q.empty()) q.pop();
        query(root);
        ans=q.top();
        printf("%d\n",ans.id);
    }
}