leetcode-中等题-450.删除二叉搜索树中的节点

https://leetcode-cn.com/problems/delete-node-in-a-bst/
https://leetcode-cn.com/problems/delete-node-in-a-bst/
这里需要考虑的条件挺多的,但实际码起来好像也不算太难,最后注意cur指针的变化会使得后面return的时候会丢失值,所以使用rootreturn

//  首先是要寻找key所在位置,BFT可以使用中序遍历
//  考虑递归函数:
//  传入参数:root key
//  返回:root
//  终止条件:遇到空节点,则返回
//  处理逻辑:(当遍历到cur.val = val时)
//      1.左右都空,直接cur == null
//      2.左不空右空, cur = cur.left;
//      3.左空右不空, cur = cur.right;
//      4.左右都不空, 左边是小的,右边是大的,所以把小的放到大那边的最小处
//         temp = cur.left;
//         cur = cur.right;
//         while(cur.left !=null) cur = cur.left; cur.left = temp;
//     左节点递归
//     右节点递归
        
class Solution {
    TreeNode cur;
    public TreeNode deleteNode(TreeNode root, int key) {
        // System.out.println(root.val);
        cur = root;
        if(cur == null) return null;
        if(cur.val == key){
            if(cur.left == null && cur.right == null) return null;
            else if(cur.left != null && cur.right == null) return cur.left;
            else if(cur.left == null && cur.right != null) return cur.right;
            else{
                TreeNode temp = cur.left;
                cur = cur.right;
                //System.out.println(cur.val);
                while(cur.left != null) cur = cur.left; 
                cur.left = temp;
                return root.right;
            }
        }
            root.right = deleteNode(root.right, key);
            root.left = deleteNode(root.left, key);
        return root;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值