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

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

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

力扣题目链接(opens new window)

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

例如，给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]

示例 1:

• 输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
• 输出: 6
• 解释: 节点 2 和节点 8 的最近公共祖先是 6。

示例 2:

• 输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
• 输出: 2
• 解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。

说明:

• 所有节点的值都是唯一的。
• p、q 为不同节点且均存在于给定的二叉搜索树中。

思路:递归法

二叉树搜索树任意两个结点的最近祖先
可以用236. 二叉树的最近公共祖先，这道题的代码解决
但这里采用二叉搜索树的特性来解决
二叉树搜索树的【祖先结点】val大于等于左节点，小于等于右节点
那么为什么会存在等于的情况呢？ 原因是p本身就是q的祖先节点，那么p肯定等于自身呀
那么对二叉树遍历，遇到的第一个结点val区间在【p,q】之间的就是祖先结点
而递归遍历顺序，本题就不涉及到 前中后序了（这里没有中节点的处理逻辑，遍历顺序无所谓了）
过程如下
1.如果根节点的值大于p和q,那么走左子树
2.如果根结点的值小于p和q，那么走右子树
3.如果根结点的值在P,Q的闭区间，那么就返回

代码如下

// 时间复杂度o(n)
// 空间复杂度o(n) 
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    return travel(root, p, q);
}

public TreeNode travel(TreeNode root, TreeNode p, TreeNode q) {
    // 如果根节点的值大于p,q
    if (root.val > p.val && root.val > q.val){
        return travel(root.left, p, q);
    }else if(root.val < p.val && root.val < q.val){
        return travel(root.right, p, q);
    }else
        return root;
}

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

力扣题目链接(opens new window)

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。

提示：

• 给定的树上的节点数介于 0 和 10^4 之间
• 每个节点都有一个唯一整数值，取值范围从 0 到 10^8
• -10^8 <= val <= 10^8
• 新值和原始二叉搜索树中的任意节点值都不同

思路:递归法

向二叉平衡树插入结点，只需要遍历二叉平衡树
比当前节点大，走右子树
比当前结点小，走左子树
最后遇叶子结点中止即可

代码如下

// 时间复杂度o(n)
// 空间复杂度o(n)
public TreeNode insertIntoBST(TreeNode root, int val) {
    return travel(root,val);
}

public TreeNode travel(TreeNode root, int val) {
    if (root == null) {
        TreeNode treeNode = new TreeNode(val, null, null);
        return treeNode;
    }

    if (val > root.val) {
        TreeNode node = travel(root.right, val);
        if (node.val == val)
            root.right = node;
    }
    if (val < root.val) {
        TreeNode node = travel(root.left, val);
        if (node.val == val)
            root.left = node;
    }
    return root;

}

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

力扣题目链接(opens new window)

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

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点； 如果找到了，删除它。 说明： 要求算法时间复杂度为 $O(h)$，h 为树的高度。

示例:

思路:递归法

第一种情况：没找到删除的节点，遍历到空节点直接返回了
找到删除的节点
第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点
第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点
第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上，返回删除节点右孩子为新的根节点。
递归法三要素
1.方法入参和返回值 入参：TreeNode root int key 返回值 root
2.中止条件 遍历的结点为空
3.核心逻辑:当遇到删除结点
1.如果左右子树均为空，那么返回Null
2.如果右子树为空，左子树不为空，那么直接返回左子树
3.如果右子树不为空，左子树为空，那么直接返回右子树
4.如果左右子树均不为空，【右子树最左边的结点】

代码如下

// 时间复杂度o(n)
// 空间复杂度o(n)
public TreeNode deleteNode(TreeNode root, int key) {
    return travel(root, key);

}

public TreeNode travel(TreeNode root, int key) {
    if (root == null)
        return null;

    if (root.val > key) {
        root.left = deleteNode(root.left, key);
    }
    if (root.val < key) {
        root.right = deleteNode(root.right, key);
    }

    if (root.val == key && root.right == null && root.left == null) {// 如果左右子树均为空，那么返回Null
        return null;
    } else if (root.val == key && root.right == null) {// 右子树为空，左子树不为空，那么直接返回左子树
        return root.left;
    } else if (root.val == key && root.left == null) {// 右子树不为空，左子树为空，那么直接返回右子树
        return root.right;
    } else if (root.val == key) {
        TreeNode rightNode = root.right;
        while (rightNode.left != null) {
            rightNode = rightNode.left;
        }
        rightNode.left = root.left;
        root = root.right;

        return root;
    }
    return root;
}