在KD-Tree的构建例程中讨厌的bug

时间:2013-07-29 15:42:49

标签: c++ kdtree

我目前正在尝试在2维(纬度和经度)中构建KD树来查询最近的坐标。我将坐标(ID,纬度,经度)推入向量中,并在我的KD树的构造函数中传递此向量。

但是当我在构造函数的末尾设置断点时(在返回递归构建调用之后)我的root-item在其左右指针中只包含垃圾。在构建例程期间,似乎它们包含相当长的时间正确的值,但在某些时候它们会丢失...有没有人看到我在递归例程中明显犯的错误?

#include "stdafx.h"
#include "KDTree.h"


KDTree::KDTree(void)
{
}

KDTree::KDTree(vector<Coordinate*> c)
{
    coordinates = c;
    root = &KDItem();
    build(root, 0, 0, c.size() - 1);
}


KDTree::~KDTree(void)
{
}

void KDTree::build(KDItem* item, int depth, int startIndex, int stopIndex)
{
    int coordinateCount = stopIndex - startIndex + 1;
    if (coordinateCount == 1)
    {
        (*item).left = NULL;
        (*item).right = NULL;
        (*item).value = NULL;
        (*item).leaf = coordinates[startIndex];
        return;
    }

    (*item).left = &KDItem();
    (*item).right = &KDItem();

    int medianIndex = 0;
    float median = 0;

    if (depth % 2 == 0)
    {
        sort(coordinates.begin() + startIndex, coordinates.begin() + stopIndex + 1, Coordinate::sortByLatitude);
        Coordinate::findLatitudeMedian(&coordinates, startIndex, stopIndex, &median, &medianIndex);
    }
    else
    {
        sort(coordinates.begin() + startIndex, coordinates.begin() + stopIndex + 1, Coordinate::sortByLongitude);
        Coordinate::findLongitudeMedian(&coordinates, startIndex, stopIndex, &median, &medianIndex);
    }

    (*item).value = median;
    build((*item).left, depth+1,startIndex, medianIndex-1);
    build((*item).right, depth+1,medianIndex,stopIndex);
}

int KDTree::findNearestNodeIndex(float latitude, float longitude)
{
    KDItem* item = findNearestItem(latitude, longitude, root);
    return (*(*item).leaf).Index();
}

KDItem* KDTree::findNearestItem(float firstVal, float secondVal, KDItem* item)
{
    if ((*item).value == NULL) return item;

    if (firstVal < (*item).value)
    {
        return findNearestItem(secondVal, firstVal, (*item).left);
    }
    else
    {
        return findNearestItem(secondVal, firstVal, (*item).right);
    }
}

1 个答案:

答案 0 :(得分:2)

KDItem *item = &KDItem()不是分配新对象的有效方法。这个KDItem将存在于堆栈中,直到它超出范围(例如,在函数返回时)。试试

(*item).left = new KDItem();

确保delete析构函数中的对象。此外,写item->left代替(*item).left

更加惯用(和可读)