删除`BST`的节点

删除BST的节点

1.题目描述:

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

2.思路

1.找到需要删除的节点,找到,删除 时间复杂度**O(h)**h为树高
2.调整BST

递归法的涉及单层逻辑

• 确定单层逻辑(主要有五种情况)
  找到删除节点,直接返回
  • 找到删除节点
    • 2.左右孩子为空(叶子节点),直接删除,返回null为根节点
    • 3.删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
    • 4.删除右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
    • 5.左右孩子都不为空,将删除节点左子树头节点,放到删除节点的右子树最左面节点的左孩子上,返回删除节点右孩子为新的根节点

3.代码(递归思路)

严格遵守递归三部曲!!!

package 二叉树;

public class 删除BST的节点 {
    /**
     * 给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。
     * 思路:
     * 1.找到需要删除的节点,找到,删除   时间复杂度O(h)h为树高
     * 2.调整BST
     */
    //1.递归法
    //确定递归函数参数和返回值
    public TreeNode delNode(TreeNode root,int key){
        //第一种:没有找到删除节点,直接返回
        //确定终止条件为空返回
        if(root==null) return root;
        /**
         * 确定单层逻辑(主要有五种情况)
         * 1.未找到删除节点,直接返回
         * 找到删除节点
         *      2.左右孩子为空(叶子节点),直接删除,返回null为根节点
         *      3.删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
         *      4.删除右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
         *      5.左右孩子都不为空,将删除节点左子树头节点,放到删除节点的右子树最左面节点的左孩子上,返回删除节点右孩子为新的根节点
         */
        //2
        if(root.val==key){
            if(root.left==null&&root.right==null){
                return null;
            }
            //3
            else if(root.left==null){
                TreeNode retNode=root.right;
                return retNode;
            }
            //4
            else if(root.right==null){
                TreeNode retNode=root.left;
                return retNode;
            }
            //5
            else{
                //找右子树最左面节点
                TreeNode cur=root.right;
                while(cur.left!=null){
                    cur=cur.left;
                }
                cur.left=root.left;
                TreeNode tmp=root;
                root=root.right;
                return root;
            }
        }
        if(root.val>key) root.left=delNode(root.left,key);
        if(root.val<key) root.right=delNode(root.right,key);
        return root;

    }


}

完结