leetcode 刷题_day22_二叉树08_BST 搜索 & 插入 &删除

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

思路

1. 普通二叉树需要从下往上，而 BST 可以借助其有序的特性
2. 如果 中间节点是 q 和 p 的公共祖先，那么 中节点的数组 一定是在 [p, q]区间的。即 中节点 > p && 中节点 < q 或者 中节点 > q && 中节点 < p。
3. 所以当从上向下去递归遍历，第一次遇到 cur节点是数值在[p, q]区间中，那么cur就是 p和q的最近公共祖先。

解题方法

描述你的解题方法

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root.val > p.val && root.val > q.val) return lowestCommonAncestor(root.left, p, q);
        if (root.val < p.val && root.val < q.val) return lowestCommonAncestor(root.right, p, q);
        return root;
    }
}

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

思路

根据 BST 的特性（有序），对 BST 进行遍历，对每个节点的值与 val 进行判断，最后遇到 null 就是需要插入的地方

解题方法

描述你的解题方法

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * 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 {
    // 根据 BST 特性，递归遍历
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            TreeNode node = new TreeNode(val);
            return node;
        }

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

        return root;
    }

    
}

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

思路

前序递归遍历，每次递归返回删除了节点的树/子树根节点

解题方法

处理当前节点时，分为五种情况

1. 没找到待删除节点

待删除节点为当前节点

2. 待删除节点为叶子节点，直接删除当前节点
3. 待删除节点左子节点不空，右子节点为空，返回左子节点
4. 待删除节点左子节点为空，右子节点不空，返回右子节点
5. 带删除节点左右子节点都不为空
   5.1 返回右子节点，将左子树整体迁移到右子树最左叶子节点上，作为其左子节点 --> 左子树小于右子树的任意节点

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * 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) {
        // 终止条件 --> 找到要删除的节点 --> 五种情况 --> 执行删除逻辑
        // 1. 没找到待删除节点
        if (root == null) {
            return null;
        }
        // 待删除节点为当前节点
        if (root.val == key) {
            // 2. 待删除节点为叶子节点，直接删除当前节点
            if (root.left == null && root.right == null) {
                return null;
            }
            // 3. 待删除节点左子节点不空，右子节点为空，返回左子节点
            else if (root.left != null && root.right == null) {
                return root.left;
            }
            // 4. 待删除节点左子节点为空，右子节点不空，返回右子节点
            else if (root.left == null && root.left != null) {
                return root.right;
            }
            // 5. 带删除节点左右子节点都不为空
            else {
                // 返回右子节点，将左子树整体迁移到右子树最左叶子节点上，作为其左子节点 --> 左子树小于右子树的任意节点
                TreeNode cur = root.right;
                while (cur.left != null) {
                    cur = cur.left;
                }
                cur.left = root.left;
                return root.right;
            }
        }

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

        return root;
    }
}