代码随想录第23天 | 669. 修剪二叉搜索树、108. 将有序数组转换为二叉搜索树、538. 把二叉搜索树转换为累加树、二叉树总结

最新推荐文章于 2024-10-02 05:39:15 发布

Meter孙

最新推荐文章于 2024-10-02 05:39:15 发布

阅读量48

点赞数

文章标签： javascript leetcode 数据结构

本文链接：https://blog.csdn.net/Metersun1/article/details/131825229

版权

669. 修剪二叉搜索树

本题和删除树中的节点还是有区别的，在删除当前的节点后还需要去遍历左右子树看有没有需要删的，等于在递归中再递归。

递归三部曲：

通过递归函数的返回值来移除节点。
修剪的操作并不是在终止条件上进行的，所以就是遇到空节点返回就可以了。
如果root（当前节点）的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。如果root(当前节点)的元素大于high的，那么应该递归左子树，并返回左子树符合条件的头结点。

接下来要将下一层处理完左子树的结果赋给root->left，处理完右子树的结果赋给root->right。

/**
 * 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} low
 * @param {number} high
 * @return {TreeNode}
 */
var trimBST = function (root, low, high) {
    if (root === null) return null;
    // 找到了这样的一个节点
    if (root.val < low) {
        return trimBST(root.right, low, high);
    }
    if (root.val > high) {
        return left = trimBST(root.left, low, high);
    }
    root.left = trimBST(root.left, low, high);
    root.right = trimBST(root.right, low, high);
    return root;
};

108. 将有序数组转换为二叉搜索树

从数组中间位置取值作为节点元素，左右区间数组分别作为左子树和右子树，左右区间大小相等，自然就是平衡二叉树。

递归三部曲：

依然用递归函数的返回值来构造中节点的左右孩子。首先是传入数组，然后就是左下标left和右下标right。在构造二叉树的时候尽量不要重新定义左右区间数组，而是用下标来操作原数组。
这里定义的是左闭右闭的区间，所以当区间 left > right的时候，就是空节点了。
首先取数组中间元素的位置。如果是int mid = (left + right) / 2;，**这么写其实有一个问题，就是数值越界，例如left和right都是最大int，这么操作就越界了，在二分法中尤其需要注意！**所以要这么写：int mid = left + ((right - left) / 2);，同时要注意左闭右闭。

/**
 * 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 {number[]} nums
 * @return {TreeNode}
 */
var sortedArrayToBST = function (nums) {
    const buildTree = (arr, left, right) => {
        if (left > right) return null;
        let mid = Math.floor(left + (right - left) / 2);
        let root = new TreeNode(arr[mid]);
        root.left = buildTree(arr, left, mid - 1);
        root.right = buildTree(arr, mid + 1, right);
        return root;
    }
    return buildTree(nums, 0, nums.length - 1);
};

538. 把二叉搜索树转换为累加树

从树中可以看出累加的顺序是右中左，所以我们需要反中序遍历这个二叉树，然后顺序累加就可以了。

双指针法：本题依然需要一个pre指针记录当前遍历节点cur的前一个节点，这样才方便做累加。

/**
 * 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
 * @return {TreeNode}
 */
var convertBST = function (root) {
    let pre = 0;
    const RevererInOrder = cur => {
        if (cur) {
            RevererInOrder(cur.right);
            cur.val += pre;
            pre = cur.val;
            RevererInOrder(cur.left);
        }
    }
    RevererInOrder(root);
    return root;
};