kdtree HDU5992

STL里面的nth_element()函数

用法:nth_element(first,nth,last)

int a[maxn];

nth_element(a,a+k,a+f);

作用:在a到a+f区间内，使第k位左边的都比它小，右边的都比它大。

kdtree代码:

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
#define lowbit(x) ((x)&(-x))
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=2e5+;
typedef long long LL;
typedef unsigned long long ULL;
const LL mod=1e9+;
const ULL base=1e7+;
using namespace std;
int kdd;
struct node{
    int id,g[];
    bool operator<(const node &u)const{
       return g[kdd]<u.g[kdd];
    }
}kdt[maxn<<],data[maxn];
bool flag[maxn<<];
LL dis(node a,node b){
    return LL(a.g[]-b.g[])*(a.g[]-b.g[])+LL(a.g[]-b.g[])*(a.g[]-b.g[]);
}
void build(int l,int r,int rt,int kd){
    kd%=;
    kdd=kd;
    if(l>r) return ;
    flag[rt]=;
    int mid=(l+r)>>;
    nth_element(data+l,data+mid,data+r+);
    kdt[rt]=data[mid];
    flag[rt<<]=flag[rt<<|]=;
    if(l<=mid-){
        build(l,mid,rt<<,kd+);
    }
    if(mid+<=r){
        build(mid+,r,rt<<|,kd+);
    }
}
pair<LL,node> ans;
void query(int rt,node p,int kd){
    kd%=;
    pair<LL,node> now={dis(p,kdt[rt]),kdt[rt]};
    int x=rt<<;
    int y=rt<<|;
    if(p.g[kd]>=kdt[rt].g[kd]){
        swap(x,y);
    }
    if(flag[x])
        query(x,p,kd+);
    bool ff=;
    if(ans.fi==-){
        ff=;
        if(p.g[]>=kdt[rt].g[]){
            ans=now;
        }
    }
    else{
        if(p.g[]>=kdt[rt].g[]&&(now.fi<ans.fi||now.fi==ans.fi&&now.se.id<ans.se.id)){
            ans=now;
        }
        if(LL(kdt[rt].g[kd]-p.g[kd])*(kdt[rt].g[kd]-p.g[kd])<ans.fi){
            ff=;
        }
    }
    if(flag[y]&&ff){
        query(y,p,kd+);
    }
}
int main(){
    int t;
    scanf("%d",&t);
    int n,m;
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++){
            for(int j=;j<;j++){
                scanf("%d",&data[i].g[j]);
            }
            data[i].id=i;
        }
        build(,n,,);
        while(m--){
            node p;
            for(int i=;i<;++i){
                scanf("%d",&p.g[i]);
            }
            ans.fi=-;
            query(,p,);
            printf("%d %d %d\n",ans.se.g[],ans.se.g[],ans.se.g[]);
        }
    }
}