找到构建kd树的坐标中位数(2D) - C ++

时间:2015-08-05 15:49:44

标签: c++ median nearest-neighbor kdtree

我在2D问题中遇到了最近邻问题,我发现kd-tree是最好的解决方案。 我无法为我正在使用的结构找到一个现成的实现,所以我决定创建自己的结构。

我使用的结构是:

struct Point{
  int id;
  double x;
  double y;
};

我有将近100000点,我的问题是:每次我想要分割我的点时如何继续找到中间点,以及如何在同一时间定义左右分区?

另一个问题是:是否有更有效的方法可以继续? (耗费的时间越少)。

1 个答案:

答案 0 :(得分:0)

  

如何计算中位数?有没有更有效的方法继续进行?

我将对这两个问题给出一个答案:您可以使用std::nth_element,如下所示:

std::vector<float> v;                                   // data vector (global)
bool myfunction (int i,int j) { return (v[i]<v[j]); } 

int find_median(std::vector<int> &v_i)
{
    size_t n = v_i.size() / 2;
    nth_element(v_i.begin(), v_i.begin()+n, v_i.end(), myfunction);
    return v_i[n];
}

您还可以查看我的question了解更多信息。

  

如何在同一时间定义左右分区?

从中位数开始的每个值都属于左侧分区,而每个大于中位数的值都属于右侧分区。

由您决定中位数的相等值会在何处。只需向左或向右选择并记住您的决定。