450. Delete Node in a BST

Given a root node reference of a BST and a key, delete the node with the given key in the BST. Return the root node reference (possibly updated) of the BST.

Basically, the deletion can be divided into two stages:

  1. Search for a node to remove.
  2. If the node is found, delete the node.

Note: Time complexity should be O(height of tree).

递归的思想,如果当前节点的值>target就在左子树里递归,<target就在右子树里递归。

如果要删除当前节点的话又分两种情况:

1、当前节点左子树或右子树为空:直接将不为空的一边赋给当前节点就好。

2、找到当前右子树的最小值赋给当前节点,递归删除右子树中的最小值节点。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        
        if(!root) return NULL;
        
        if(root->val>key) root->left=deleteNode(root->left, key);
        else if(root->val<key) root->right=deleteNode(root->right, key);
        else{
            if(!root->left||!root->right) root=!(root->left)?root->right:root->left;
            else{
                auto rightMin=root->right;
                while(rightMin->left) rightMin=rightMin->left;
                root->val=rightMin->val;
                root->right=deleteNode(root->right, rightMin->val);
            }
        }
        return root;
    }
};


  • 0
    点赞
  • 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、付费专栏及课程。

余额充值