450.Delete Node in a BST

26 篇文章 2 订阅

Delete Node in a BST

在二分查找树中删除一个给定值所在的节点,随后返回删除后二叉树的根节点。

解题思路:

利用二分查找树查找该节点key,key和当前根节点的value进行比较:
(1) 如果key < < <script type="math/tex" id="MathJax-Element-64"><</script>value,说明要删除的节点在根节点的左子树上。
(2) 如果key > > <script type="math/tex" id="MathJax-Element-65">></script>value,说明要删除的节点在根节点的右子树上。
(3) 如果key == value,表明要删除的节点就是当前根节点,按照如下情况讨论:

(1) 如果root中左子树为空,那么设置删除后二叉树的根节点为root的右子树;
(2) 如果root中右子树为空,那么设置删除后二叉树的根节点的root的左子树;
(3) 如果root中左右子树都不是空,有两种方法,用root的左子树中的最大值max代替root的值,然后删除root所在左子树上的max值,把返回的节点挂在root的左子树上。或者用root的右子树中的小值min代替root的值,然后删除root所在右子树上的min值,把返回的节点挂在右子树上。

具体代码如下:

class Solution {
public:
    // find the max value's node
     TreeNode* findMin(TreeNode* root){
        while (root->right)
            root = root->right;
        return root;
    }
TreeNode* deleteNode(TreeNode* root, int key) {
    if (root == nullptr)
        return nullptr;
    if (key< root->val)
        root->left = deleteNode(root->left, key);
    else if (key>root->val)
        root->right = deleteNode(root->right, key);
    else
    {
        if (root->left == nullptr)
        {
            TreeNode* tmp = root;
            root = root->right;
            delete tmp;
        }
        else if (root->right == nullptr)
        {
            TreeNode* tmp = root;
            root = root->left;
            delete tmp;
        }
        else
        {
            // 寻找左子树上最大的节点
            TreeNode* max Node = findMin(root->left);
            root->val = minNode->val;
            root->left = deleteNode(root->left, root->val);
        }
    }
    return root;
    }
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个二叉搜索树(Binary Search Tree)的 Python 实现,包括了插入、删除、搜索和转换为列表等方法。 ```python class TreeNode: """The tree node class""" def __init__(self, val): self.val = val self.left = None self.right = None class BSTree: """The binary search tree""" def __init__(self, val): """Initialize the BSTree""" self.root = TreeNode(val) def search(self, val): """Search for a value in the tree, and return the tree node @return: the tree node that contain the val None if val is not in the tree """ curr_node = self.root while curr_node: if curr_node.val == val: return curr_node elif curr_node.val > val: curr_node = curr_node.left else: curr_node = curr_node.right return None def insert(self, val): """Insert a value""" new_node = TreeNode(val) if not self.root: self.root = new_node return curr_node = self.root while curr_node: if curr_node.val > val: if curr_node.left: curr_node = curr_node.left else: curr_node.left = new_node return else: if curr_node.right: curr_node = curr_node.right else: curr_node.right = new_node return def delete(self, val): """Delete a value from the tree""" def find_min_node(node): while node.left: node = node.left return node def delete_node(node, val): if not node: return None if node.val == val: if not node.left and not node.right: return None if not node.left: return node.right if not node.right: return node.left min_node = find_min_node(node.right) node.val = min_node.val node.right = delete_node(node.right, min_node.val) elif node.val > val: node.left = delete_node(node.left, val) else: node.right = delete_node(node.right, val) return node self.root = delete_node(self.root, val) def toList(self): """Convert tree values into a list in in-order traversal""" res = [] def inorder_traversal(node): if not node: return inorder_traversal(node.left) res.append(node.val) inorder_traversal(node.right) inorder_traversal(self.root) return res ``` 使用方法: ```python bst = BSTree(5) bst.insert(3) bst.insert(7) bst.insert(2) bst.insert(4) bst.insert(6) bst.insert(8) print(bst.toList()) # [2, 3, 4, 5, 6, 7, 8] bst.delete(5) print(bst.toList()) # [2, 3, 4, 6, 7, 8] node = bst.search(7) print(node.val) # 7 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值