代码随想录算法训练营第二十三天| 669. 修剪二叉搜索树、108. 将有序数组转换为二叉搜索树、538. 把二叉搜索树转换为累加树。

669. 修剪二叉搜索树

题目链接:https://leetcode.cn/problems/trim-a-binary-search-tree/

题目要求:

给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。

示例 1:

输入:root = [1,0,2], low = 1, high = 2

输出:[1,null,2]

总结:如果该节点的值小于区间的最小值,则说明该节点与其左子树都不能要了,直接删除,返回以右孩子为根再开一个这个函数的结果,最后返回的也就是以这个右孩子为根的树,相当于间接把原本的该节点和其左子树删除掉了,如果该节点的值大于区间的最大值,则该节点包括其右子树的所有节点不能要了,直接删除,以左孩子为根,再开一个这个函数,同样相当于把该节点和其右子树都删除掉了,符合左右区间的话,则该节点不用删除,像其左右子树递归查看,最终返回根节点。

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if(root == null) return root;
        //如果该节点的值小于区间的最小值,则说明该节点与其左子树都不能要了,直接删除,以右孩子为根再开一个这个函数
        if(root.val < low){
            return trimBST(root.right,low,high);
        }
        //如果该节点的值大于区间的最大值,则该节点包括其右子树的所有节点不能要了,直接删除,以左孩子为根,再开一个这个函数。
        if(root.val > high){
            return trimBST(root.left,low,high);
        }
        //在区间里,则以此向左右递归
        root.left = trimBST(root.left,low,high);
        root.right = trimBST(root.right,low,high);
        return root;

    }
}

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

题目链接:https://leetcode.cn/problems/convert-sorted-array-to-binary-search-tree/description/

题目要求:

给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1:

输入:nums = [-10,-3,0,5,9]

输出:[0,-3,9,-10,null,5]

解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:

总结:将有序数组转化为平衡二叉树,构造二叉树想前序遍历,平衡二叉树,那么就每次从数组中挑出中间位置的元素构造节点,然后根据中间元素位置,划分数组前后区间,向左向右依次递归,构成该节点的左右子树,切分出的左右数组,也是先看如果只剩下一个元素,直接构造节点即可,也不用挑出中间了,如果大于1个节点,则先挑出中间位置的节点构造中间节点,然后再向左向右去递归,最终返回root即可。

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        //区间左闭右开
        return sortedArrayToBST(nums,0,nums.length);
    }
    private TreeNode sortedArrayToBST(int[] nums,int start,int end){
        if(end - start < 1){
            return null;
        }
        if(end - start == 1){
            return new TreeNode(nums[start]);
        }
        int mid = start + (end - start)/2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(nums,start,mid);
        root.right = sortedArrayToBST(nums,mid + 1,end);
        return root;
    }
}

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

题目链接:https://leetcode.cn/problems/convert-bst-to-greater-tree/description/

题目要求:

给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下,二叉搜索树满足下列约束条件:

  • 节点的左子树仅包含键 小于 节点键的节点。

  • 节点的右子树仅包含键 大于 节点键的节点。

  • 左右子树也必须是二叉搜索树。

注意:本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同

示例 1:

输入:[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]

输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

总结:根据题目要求,是将遍历的每个节点值赋值为该节点本身的值以及大于该节点值的所有值的和,也就是说要将其该节点与其右子树的所有节点都遍历出来再相加,感觉很复杂,但是利用二叉搜索树的性质,中序遍历即可想成有序数组,累计这些值就简单很多,但是又是优先算右子树的节点值之和,所有由左中右小小变化为右中左的顺序,这样相当逆序遍历,从大到小的顺序,这样我们去遍历,就可以通过一个sum变量,由大到小的去统计这些值的和,比如【1,2,3,4,5】,我们先从5开始遍历,用sum统计5的值,再遍历到4,将4+之前统计的sum,也就是5,加上之后再赋值给4位置这个节点,用sum也加上该节点4,这样依次遍历,也就相当于每次遍历到一个节点,就将本节点的值+大于其值的所有节点值的和(sum),再赋值给该节点。

class Solution {
    int sum;
    public TreeNode convertBST(TreeNode root) {
        sum = 0;
        convertBST1(root);
        return root;
    }
    private void convertBST1(TreeNode root){
        if(root == null) return;
        convertBST1(root.right);
        sum = sum + root.val;
        root.val = sum;
        convertBST1(root.left);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值