我想使用三元搜索树中的键删除特定节点。这在大多数情况下工作得很好,但在我的一些测试集中,没有中间孩子的节点也不存储值,这不应该发生。
我尝试了在网上找到的不同方法,但几乎所有这些方法都使树处于脏状态,这使搜索变得很麻烦,因为您需要检查找到的叶子是否确实具有值,而这不应该发生。
这是我的相关代码
private boolean hasChildren(Node x) {
return (x.left != null || x.mid != null || x.right != null);
}
private void reHang(Node x) {
if (hasChildren(x)) {
if (x.left != null) {
x.parent.mid = x.left;
x.left.right = x.right;
} else if (x.right != null) {
x.parent.mid = x.right;
}
}
}
private boolean remove(Node x, String key, int d) {
if (x == null) return false;
char c = key.charAt(d);
if (c < x.c) {
if (!remove(x.left, key, d)) {
x.left = null;
}
} else if (c > x.c) {
if (!remove(x.right, key, d)) {
x.right = null;
}
} else if (d < key.length() - 1) {
if (!remove(x.mid, key, d + 1)) {
x.mid = null;
if (x.val != null) return true;
}
} else {
x.val = null;
}
reHang(x);
return hasChildren(x);
}
private class Node
{
private Value val;
private char c;
private Node left, mid, right, parent;
}
具体来说,我用于前缀查找的这个函数会出现问题:
private Node getMidPath(Node x) {
if (x.left != null) return getMidPath(x.left);
if (x.mid == null) return x;
return getMidPath(x.mid);
}
每个没有 x.mid 的节点都应该有一个 x.val 集,删除后并不总是这样,这意味着我有脏节点。
任何帮助将不胜感激。