Question

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

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);
}

如何在二进制搜索树中的每个节点上设置位置？

1 个答案: