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

今日内容

 

 

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

思路：递归＋回溯

/*
 * @lc app=leetcode.cn id=235 lang=javascript
 *
 * [235] 二叉搜索树的最近公共祖先
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */

/**
 * @param {TreeNode} root
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {TreeNode}
 */
var lowestCommonAncestor = function(root, p, q) {

    if (!root) return root;
    if (root.val === p.val || root.val === q.val) return root;

    // 单层逻辑
    let left = lowestCommonAncestor(root.left, p, q);
    let right = lowestCommonAncestor(root.right, p, q);

    if (left !== null && right !== null) return root;
    else if (left === null && right !== null) return right;
    else if (left !== null && right === null) return left;
    else return null;

};

 

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

 

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

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

 思路：递归

我的思路：插入后调整二叉树，其实本题不需要这么复杂（如果不要求平衡二叉树的话，只需要往叶子节点插入就好了）

理论上想到怎么做，但是为什么代码这么简洁？这段代码在哪里进行了插入，又在哪里进行了调整二叉树？？？

视频讲解

/*
 * @lc app=leetcode.cn id=701 lang=javascript
 *
 * [701] 二叉搜索树中的插入操作
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} val
 * @return {TreeNode}
 */
var insertIntoBST = function(root, val) {
    //
    if (!root) return new TreeNode(val); // 返回插入的节点

    if (root.val > val) { // 如果插入值小于当前节点的值，递归插入到左子树
        root.left = insertIntoBST(root.left, val); //插入到左叶子节点下面
    } else { // 如果插入值大于等于当前节点的值，递归插入到右子树
        root.right = insertIntoBST(root.right, val);//插入到右叶子节点下面

 } 30 31 return root; // 返回调整后的树 32 33 };

这道题简单粗暴的方法是：只放在叶子节点上，不考虑再中间插入并重构，如果是平衡二叉树的话就需要了！

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

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

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

1. 首先找到需要删除的节点；
2. 如果找到了，删除它。

 考虑到左右子树都不为空的情况下，继续找到右子树的最小节点来替换成为根节点，且要将最小节点从右子树中删除。

 

/*
 * @lc app=leetcode.cn id=450 lang=javascript
 *
 * [450] 删除二叉搜索树中的节点
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} key
 * @return {TreeNode}
 */
var deleteNode = function(root, key) {

    findMin = function(root) {
            while (root.left) {
                root = root.left;
            }
            return root;
        }
        // 终止条件是：找到节点并删除
    if (!root) return null; // 
    if (root.val === key) {
        if (root.left && root.right) { // 如果要删除的节点有左右子树
            let minNode = findMin(root.right); // 找到右子树中最小的节点
            root.val = minNode.val; // 用右子树中最小节点的值替换当前节点的值
            root.right = deleteNode(root.right, minNode.val); // 删除右子树中最小节点
        } else if (root.left && root.right === null) { // 如果要删除的节点只有左子树，直接返回左子树
            return root.left;
        } else if (root.left === null && root.right) { // 如果要删除的节点只有右子树，直接返回右子树
            return root.right;
        } else { //左右节点均为空
            return null;
        }
    }

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

    return root;

};