给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。
示例:
递归查找节点删除
找到节点后有4种情况
1、节点左孩子为空,则让节点等于右孩子
2、节点右孩子为空,则让节点等于左孩子
3、左右孩子都为空,则为null,已经被1包括,在判断1的时候,右孩子为空赋值节点效果一营。
4、左右还自己都不为空,则找到右子树中最小的节点与节点替换。这样成为,删除这个最小节点,递归继续即可。(最后直至删除叶子节点时结束)
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if(root == null) 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.right == null){
root = root.left;
}else if(root.left == null){
root = root.right;
}else{//寻找最小节点然后与root值替换
TreeNode temp = root.right;
while(temp.left != null){//找到最小节点。
temp = temp.left;
}
root.val = temp.val;
//替换后,需要删除找到的那个最小值节点,可以继续递归
root.right= deleteNode(root.right,temp.val);
}
}
return root;
}
}