代码随想录算法训练营第22天

今日任务

● 235. 二叉搜索树的最近公共祖先 ● 701.二叉搜索树中的插入操作 ● 450.删除二叉搜索树中的节点

链接：https://docs.qq.com/doc/DUHplVUp5YnN1bnBL (opens in a new tab)

235. 二叉搜索树的最近公共祖先

自己想法

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        return invert(root, p, q);
    }
    // 1.返回值：当前找到的公共祖先
    public TreeNode invert(TreeNode root, TreeNode p, TreeNode q) {
        // 2.结束条件
        if(root == p || root == q || root == null) return root;
        // 3.递归逻辑
        if(root.val < p.val && root.val < q.val) return invert(root.right, p, q);
        else if(root.val > p.val && root.val > q.val) return invert(root.left, p, q);
        else {
            // 此时root.val一定介于p和q之间
            return root;
        }
        
    }
}

701.二叉搜索树中的插入操作

自己想法

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        // 处理一下空树情况
        if(root == null) root = new TreeNode(val);
        else invert(root, val);
        return root;
    }
    // 1.不修改root，无返回值
    public void invert(TreeNode node, int val) {
        // 2.返回条件：无，因为不是每次都进递归
        // 3.递归逻辑
        if(node.val > val) {
            if(node.left != null) invert(node.left, val);
            else {
                TreeNode tmp = new TreeNode(val);
                node.left = tmp;
            }
        }
        else if (node.val < val) {
            if(node.right != null) invert(node.right, val);
            else {
                TreeNode tmp = new TreeNode(val);
                node.right = tmp;
            }
        }
        else return;
    }
}

450.删除二叉搜索树中的节点

题解想法

  1. 说到递归函数的返回值，可以通过递归返回值删除节点。

  2. 二叉搜索树中删除节点遇到的五种情况：

  • 第一种情况：没找到删除的节点，遍历到空节点直接返回了

  • 找到删除的节点

    • 第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点

    • 第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点

    • 第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点

    • 第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上， 返回删除节点右孩子为新的根节点。

/**
 * 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 root;
        if(root.val == key) {
            if(root.left == null) return root.right;
            else if(root.right == null) return root.left;
            else {
                TreeNode tmp = root.right;
                while(tmp.left != null) tmp = tmp.left;
                tmp.left = root.left;
                root = root.right;
                return root;
            }
        }
        else if(root.val > key) {
            // 接住删掉后的子树root
            root.left = deleteNode(root.left, key);
            return root;
        }
        else {
            // 接住删掉后的子树root
            root.right = deleteNode(root.right, key);
            return root;
        }
    }
}