Toasobi
删除二叉搜索树的节点
本文最后更新于2023年09月14日,已超过480天没有更新。如果文章内容或图片资源失效,请留言反馈,我会及时处理,谢谢!
根据二叉搜索树的性质:
- 如果目标节点大于当前节点值,则去右子树中删除;
- 如果目标节点小于当前节点值,则去左子树中删除;
如果目标节点就是当前节点,分为以下三种情况:
- 其无左子:其右子顶替其位置,删除了该节点;
- 其无右子:其左子顶替其位置,删除了该节点;
- 其左右子节点都有:其左子树转移到其右子树的最左节点的左子树上,然后右子树顶替其位置,由此删除了该节点。
第三种情况图示如下:
代码如下:
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if (root == null)return null;
if (key > root.val)
root.right = deleteNode(root.right, key); // 去右子树删除
else if(key < root.val)
root.left = deleteNode(root.left, key); // 去左子树删除
else { // 当前节点就是要删除的节点
if (root.left == null) return root.right; // 情况1,欲删除节点无左子
else if (root.right == null) return root.left; // 情况2,欲删除节点无右子
else if (root.left!=null && root.right !=null){ // 情况3,欲删除节点左右子都有
TreeNode node = root.right;
while (node.left != null) // 寻找欲删除节点右子树的最左节点
node = node.left;
node.left = root.left; // 将欲删除节点的左子树成为其右子树的最左节点的左子树
root = root.right; // 欲删除节点的右子顶替其位置,节点被删除
}
}
return root;
}
}