删除二叉搜索树中的节点
二叉搜索树(Binary Search Tree,简称BST)是一种常见的数据结构,它具有以下性质:
- 对于任意节点,其左子树中的所有节点的值都小于该节点的值。
- 对于任意节点,其右子树中的所有节点的值都大于该节点的值。
- 左子树和右子树也都是二叉搜索树。
在二叉搜索树中删除节点是一个常见的操作,我们需要保证删除节点后的树仍然满足二叉搜索树的性质。本文将介绍如何在二叉搜索树中删除节点的算法和实现。
算法思路
删除二叉搜索树中的节点可以分为以下几个步骤:
- 首先,我们需要找到要删除的节点。从根节点开始,比较要删除的节点的值与当前节点的值,根据二叉搜索树的性质,如果要删除的节点的值小于当前节点的值,则继续在左子树中查找;如果要删除的节点的值大于当前节点的值,则继续在右子树中查找;如果要删除的节点的值等于当前节点的值,则找到了要删除的节点。
- 如果找到了要删除的节点,我们需要考虑以下几种情况:
- 如果要删除的节点是叶子节点(没有左子树和右子树),直接将其删除即可。
- 如果要删除的节点只有一个子节点(左子树或右子树),将其子节点替换为要删除的节点。
- 如果要删除的节点有两个子节点,我们可以选择以下两种方法之一:
- 找到要删除节点右子树中的最小节点,将其值替换到要删除的节点上,然后递归地删除右子树中的最小节点。
- 找到要删除节点左子树中的最大节点,将其值替换到要删除的节点上,然后递归地删除左子树中的最大节点。
- 最后,返回更新后的二叉搜索树的根节点引用。
代码实现
方法一:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if (root == null) {
return