BST 删除 - 删除“tmp”会导致树丢失

Question

debug result

附上我试图在 bst 中删除节点的代码。

如果我想删除节点1，当在“if (del_node->l_ == NULL)”中指定tmp = del，并删除tmp时，del也被删除，树数据丢失。我该如何解决这个问题？

示例树：

  3
 / \
1   5
 \ 
  2

• 为简单起见，所有数据成员和函数都声明为 public。

    void BST::DeleteNode(int data) {
        BinaryTreeNode* &del_node = BST_Search(head_, data);
        if (!del_node->l_ && !del_node->r_)
        {
            delete del_node;
            del_node = nullptr;
            return;
        }
        if (del_node->l_ == NULL)
        {
            BinaryTreeNode* tmp = del_node;
            del_node = del_node->r_;
            tmp = nullptr;
            delete tmp;
            return;
        }
        if (del_node->r_ == NULL)
        {
            BinaryTreeNode* tmp = del_node;
            del_node = del_node->l_;
            delete tmp;
            return;
        }
        else
        {
            del_node->data_ = smallestRightSubTree(del_node->r_);
        }
  
    }
  
    int BST::smallestRightSubTree(BinaryTreeNode* rightroot)
    {
        // if rightroot has no more left childs
        if (rightroot && !rightroot->l_)
        {
            int tmpVal = rightroot->data_;
            BinaryTreeNode* tmp = rightroot;
            rightroot = rightroot->r_;
            delete tmp;
            return tmpVal;
        }
        return smallestRightSubTree(rightroot->l_);
    }
    int main()
    {
        BST bst;
        bst.BST_Insert(bst.head_, 3);
        bst.BST_Insert(bst.head_, 5);
        bst.BST_Insert(bst.head_, 1);
        bst.BST_Insert(bst.head_, 2);
        bst.DeleteNode(1);
        return 0;
    }
  

感谢您的帮助！ 编辑：这是条件“if(del->l = null)”中的“del_node = del_node->r_)”行之后 tmp 和 del_node 的样子

void BST::BST_Insert(BinaryTreeNode*& head, int data) {
if (head == nullptr) {
    head = new BinaryTreeNode(data, nullptr, nullptr);
    return;
}
if (data > head->data_) {
    BST_Insert(head->r_, data);
}
else {
    BST_Insert(head->l_, data);
}

}

BinaryTreeNode* BST::BST_Search(BinaryTreeNode* root, int key) {
    if (root == nullptr || root->data_ == key)
        return root;
    if (key > root->data_)
        return BST_Search(root->r_, key);
    return BST_Search(root->l_, key);
}

Answer 1

如果您的BST_Search 按值返回BinaryTreeNode*，那么BinaryTreeNode* &del_node = BST_Search(head_, data); 中的实际引用是什么？它分配一个新的临时并引用它。您可能希望它引用保存树中指针的变量，以便您可以修改树。

您的 BST_Search 必须如下所示：

BinaryTreeNode*& BST::BST_Search(BinaryTreeNode*& root, int key) {
    if (root == nullptr || root->data_ == key)
        return root;
    if (key > root->data_)
        return BST_Search(root->r_, key);
    return BST_Search(root->l_, key);
}

我无法检查这是否真的有效，因为您没有提供一个独立的可编译示例。但是有些类似的东西。