训练营day22|二叉树（二叉搜索树的最近公共祖先|插入操作|删除节点）

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

二叉搜索树是有序数组，找到cur节点在p和q结点值中间，则cur节点则是结果

class Solution {
    /**
     * 二叉搜索树的最近公共祖先
     *
     * @param root
     * @param p
     * @param q
     * @return
     */
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        return dou(root, p, q);
    }

    public TreeNode dou(TreeNode root, TreeNode p, TreeNode q) {
        //确认递归函数的出口
        if (root == null) {
            return null;
        }
        //确认单层递归的逻辑
        if (root.val > p.val && root.val > q.val) {//根节点的值比p，q的值都大，在根节点的左子树进行遍历
            TreeNode left = dou(root.left, p, q);
            if (left != null) {
                return left;
            }
        }
        if (root.val < p.val && root.val < q.val) {
            TreeNode right = dou(root.right, p, q);
            if (right != null) {
                return right;
            }
        }
        return root;
    }
}

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

class Solution {
    /**
     * 二叉搜索树中插入结点
     *
     * @param root
     * @param val
     * @return
     */
    public TreeNode insertIntoBST(TreeNode root, int val) {
        //递归函数出口
        if (root == null) {
            return new TreeNode(val);
        }
        //递归函数单层逻辑
        if (root.val < val) {
            root.right = insertIntoBST(root.right, val);
        }
        if (root.val > val) {
            root.left = insertIntoBST(root.left, val);
        }
        return root;
    }
}

---

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

class Solution {
    /**
     * 二叉搜索树删除结点
     *
     * @param root
     * @param key
     * @return
     */
    public TreeNode deleteNode(TreeNode root, int key) {
        return delete(root, key);
    }

    public TreeNode delete(TreeNode root, int key) {
        if (root == null) {
            return null;
        }
        if (root.val > key) {
            root.left = delete(root.left, key);
        }
        if (root.val < key) {
            root.right = delete(root.right, key);
        } else if (root.val == key) {
            if (root.left == null && root.right == null) {
                return null;
            } else if (root.left == null) {
                return root.right;
            } else if (root.right == null) {
                return root.left;
            } else {
                TreeNode temp = root.right;
                while (temp.left != null) {
                    temp = temp.left;
                }
                root.val = temp.val;
                root.right = delete(root.right, temp.val);
            }
        }
        return root;
    }
}