代码随想录算法训练营第二十二天 | 235. 二叉搜索树的最近公共祖先、701.二叉搜索树中的插入操作、450.删除二叉搜索树中的节点

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

题目链接：https://leetcode.cn/problems/lowest-common-ancestor-of-a-binary-search-tree/
文档讲解：https://programmercarl.com/0235.%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E7%9A%84%E6%9C%80%E8%BF%91%E5%85%AC%E5%85%B1%E7%A5%96%E5%85%88.html
视频讲解：https://www.bilibili.com/video/BV1Zt4y1F7ww

思路

• 如果当前节点的值大于p和q，说明p和q在当前节点的左子树里，就接着往左找；
• 如果当前节点的值小于p和q，说明p和q在当前节点的右子树里，就接着往右找；
• 如果当前节点的值在p和q中间，那么当前节点就是p和q的最小公共祖先。

代码

递归法

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

迭代法

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

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

题目链接：https://leetcode.cn/problems/insert-into-a-binary-search-tree/
文档讲解：https://programmercarl.com/0701.%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E4%B8%AD%E7%9A%84%E6%8F%92%E5%85%A5%E6%93%8D%E4%BD%9C.html
视频讲解：https://www.bilibili.com/video/BV1Et4y1c78Y

思路

• 自己的迭代思路：当遍历到某个节点的左或右为空，说明就是插入节点的位置。
• 学习的递归法：如果当前节点为空，直接插入，返回插入节点；如果当前节点大于插入节点，向左遍历，返回插入后的左子树；如果当前节点小于插入节点，向右遍历，返回插入后的右子树。

代码

自己的迭代思路

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) return new TreeNode(val);
        TreeNode insertNode = new TreeNode(val);
        TreeNode cur = root;
        while (cur != null) {
            if (val > cur.val) {
                if (cur.right != null) cur = cur.right;
                else {
                    cur.right = insertNode;
                    break;
                }
            } else if (val < cur.val) {
                if (cur.left != null) cur = cur.left;
                else {
                    cur.left = insertNode;
                    break;
                }
            }
        }
        return root;
    }
}

学习的递归法

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) return new TreeNode(val);
        if (val < root.val) root.left = insertIntoBST(root.left, val);
        else if (val > root.val) root.right = insertIntoBST(root.right, val);
        return root;
    }
}

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

题目链接：https://leetcode.cn/problems/delete-node-in-a-bst/
文档讲解：https://programmercarl.com/0450.%E5%88%A0%E9%99%A4%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E4%B8%AD%E7%9A%84%E8%8A%82%E7%82%B9.html
视频讲解：https://www.bilibili.com/video/BV1tP41177us

思路

• 有五种情况：
  • 没找到要删除的点
  • 删除的点是叶子节点（左右都为空），直接删除
  • 左不空右空，返回左节点（相当于跳过当前节点，把左子树接到父节点上）
  • 左空右不空，返回右节点
  • 左右都不空：将左子树接到右子树最左边的节点上（右子树的最小值的左子树）

代码

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 cur = root.right;
        while (cur.left != null) {
          cur = cur.left;
        }
        cur.left = root.left;
        root = root.right;
        return root;
      }
    }
    if (root.val > key) root.left = deleteNode(root.left, key);
    if (root.val < key) root.right = deleteNode(root.right, key);
    return root;
  }
}