如何为二叉搜索树中的每个节点设置位置?

Question

我想构建一个BST并按升序为每个节点设置一个位置。例如，如果二叉搜索树包含 3、6、2、4，则节点的位置应为 pos1-2、pos2-3、pos3-4、pos4- 6。 这是我的插入BST方法

TreeNode insertBST(TreeNode parent, int value) {
if (parent == null) {
parent = new TreeNode(value);
return parent;
}
if (value <= parent.value) {
parent.leftChild = this.insertBST(parent.leftChild, value);
} else if (value >= this.root.value) {
parent.rightChild = this.insertBST(this.root.rightChild, value);
}
return parent;
}

这是我想要设置位置的inOrderTraverseBST方法。

int count = 0;
public void inOrderTraverseBST(TreeNode root) {
if (root == null) {
return;
}
this.inOrderTraverseBST(root.leftChild);
root.position = this.count;
this.count++;
this.inOrderTraverseBST(root.rightChild);
root.position = this.count;
}

但是inOrderTraversBST方法是错误的。所以我想知道如何编写 inOrderTraverseBST 方法的方法，以便将所有位置设置为节点。

Answer 1

只需删除最后一行。有了它，您可以在遍历节点的右子树后重新分配节点的位置。

简化一点

public void inOrderTraverseBST(TreeNode root) {
if (root == null) {
return;
}
this.inOrderTraverseBST(root.leftChild);
root.position = this.count++;
this.inOrderTraverseBST(root.rightChild);
}