[模版] K/最近邻匹配（KD Tree）

最新推荐文章于 2024-04-02 18:37:54 发布

Martin20150405

最新推荐文章于 2024-04-02 18:37:54 发布

阅读量1k

点赞数

分类专栏：模板

本文链接：https://blog.csdn.net/Martin20150405/article/details/54927741

版权

本文介绍了如何利用KD Tree进行2D空间的最近邻查找，并探讨了在K维空间中的Q近邻匹配方法。 nth_element函数在此过程中起到了关键作用，类似于快速排序中选择枢轴的过程。

摘要由CSDN通过智能技术生成

nth_element相当于快排的split/choose pivot过程

2D 最近

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
const int INF=0x3f3f3f3f;
const int maxn=1e5+50;
typedef long long ll;
using namespace std;

int cmpNo;
struct Node{
    int x[2],l,r,id;
    bool operator <(const Node &b)const{
        return x[cmpNo]<b.x[cmpNo];
    }
};

ll calDis(Node &l,Node &r){
    ll dx=l.x[0]-r.x[0],dy=l.x[1]-r.x[1];
    return dx*dx+dy*dy;
}

Node p[maxn];

int Build(int l,int r,int d){
    if(l>r)return 0;
    cmpNo=d;
    int mid=l+r>>1;
    nth_element(p+l,p+mid,p+r+1);
    p[mid].l=Build(l,mid-1,1-d);
    p[mid].r=Build(mid+1,r,1-d);
    return mid;
}

ll ansDist;
int<