KD树在查找和搜索方面表现出了良好的性能,但是在删除节点时却十分麻烦。
为了降低删除的性能开销,一个自然而然的想法就是不去真正删除KD树的子节点,而是给每一个子节点打上一个标记,当遇到带标记的子节点时忽略就是。
建立哈希表,储存要删除的节点:
// Mark deleted points // node is not deleted actually. Set<Integer> hash = new HashSet<Integer>();当要删除节点时候,直接将它加入哈希表:
public boolean deletePoint(Point point) { if (isPointIn(point)) { hash.add(point.hashCode()); return true; } return false; }
在判断条件里加上许多的hash.contain(point)来实现虽然可行,但这个逻辑实在是十分复杂。
回到KD树寻找最近的节点的依据,是它和节点的距离远近。如果我们将已经标记过的节点的距离替换成一个很大(最好是无穷,但可惜不能)的数,KD树在比较的时候就会自然略过这些点:
if (hash.contains(point.hashCode())) Distance = 9999999;只要加上这一条小小的判断,就能寻找到KD树的多个最近节点了。