450. 删除二叉搜索树中的节点 - 力扣(LeetCode)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
//终止条件,找到要删除的节点就是要终止了,在终止递归的时候进行相应节点的处理
if(root == null) return null;
if(root.val==key){
//左右孩子都为空
if(root.left==null && root.right==null){
return null;
}
//左孩子为空,右不空
else if(root.left==null&&root.right!=null){
return root.right;
}
//左不空,右空
else if(root.left!=null&&root.right==null){
return root.left;
}
//左右孩子都不空
else{
//让右孩子接班
TreeNode cur = root.right;
TreeNode pre = root.left;
//把删除节点的左孩子放在右孩子的最最左下
while(cur.left!=null){
cur = cur.left;
}
cur.left = pre;
return root.right;
}
}
//递归逻辑
if(root.val > key){
root.left = deleteNode(root.left,key);
}
else if(root.val < key){
root.right = deleteNode(root.right,key);
}
return root;
}
}