打卡-算法训练营-Day20 | 235. 二叉搜索树的最近公共祖先;701. 二叉搜索树中的插入操作;450.删除二叉搜索树中的节点

53 阅读1分钟

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

leetcode链接:leetcode.cn/problems/lo…

利用二叉树的特性,如果p和q节点在当前节点的左右两边,说明当前节点一定是p和q节点的最近公共祖先

var lowestCommonAncestor = function(root, p, q) {
    if (root === null) return null;

    if (root.val > p.val && root.val > q.val) {
        return root.left = lowestCommonAncestor(root.left, p, q);
    }

    if (root.val < p.val && root.val < q.val) {
        return root.right = lowestCommonAncestor(root.right, p, q);
    }

    return root;
};

二叉搜索树中的插入操作

leetcode链接:leetcode.cn/problems/in…

根据搜索二叉树的特性,如果要在一棵二叉搜索树中插入一个节点,这个节点一定可以是叶子节点

var insertIntoBST = function(root, val) {
  // 遇到叶子节点,可以插入新的节点了
  if (!root) {
    let newNode = new TreeNode(val);
    return newNode;
  }
  if (val < root.val) {
    root.left = insertIntoBST(root.left, val);
  }
  if (val > root.val) {
    root.right = insertIntoBST(root.right, val);
  }
  return root;
};

删除二叉搜索树中的节点

leetcode链接:leetcode.cn/problems/de…

删除比插入复杂,需要考虑的情况比较多:

  1. 找不到要删除的节点
  2. 要删除的是叶子节点,节点的左边为空,右边也为空
  3. 要删除的节点左边不为空,右边为空
  4. 要删除的节点左边为空,右边不为空
  5. 要删除的节点左边不为空,右边不为空
var deleteNode = function(root, key) {
    if (root === null) return null;

    // 找到要删除的节点了
    if (root.val === key) {
        if (root.left === null && root.right === null) {
            // 删除叶子节点操作
            return null;
        } else if (root.left && root.right === null) {
            return root.left;
        } else if (root.left === null && root.right) {
            return root.right;
        } else {
            // 第5种情况,将右子树往上继承,左子树放在右子树的最左边
            let cur = root.right;
            while (cur.left) {
                cur = cur.left;
            }
            cur.left = root.left;
            return root.right;
        }
    } else if (root.val > key) {
        // 往左边找要删除的节点
        root.left = deleteNode(root.left, key);
    } else {
        // 往右边找要删除的节点
        root.right = deleteNode(root.right, key);
    }
    return root;
};