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

669. 修剪二叉搜索树

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

可以用递归的方式解决。这个问题的关键是要找到数组的中间元素作为二叉搜索树的根。然后将数组的左半部分和右半部分递归地转换为左右子树。这样,由于左右子树的元素数量相等或者相差1,可以保证得到的二叉搜索树是高度平衡的

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        // 对空数组或null进行判断
        if (nums == null || nums.length == 0) {
            return null;
        }
        // 调用递归方法来构建二叉搜索树
        return buildBST(nums, 0, nums.length - 1);
    }

    private TreeNode buildBST(int[] nums, int left, int right) {
        // 如果左索引大于右索引,则返回null,表示当前这部分数组不需要构建二叉搜索树
        if (left > right) {
            return null;
        }
        // 选择中间元素作为二叉搜索树的根节点,这样可以保证二叉搜索树是平衡的
        int mid = left + (right - left) / 2;
        TreeNode node = new TreeNode(nums[mid]);
        // 递归地构建左子树
        node.left = buildBST(nums, left, mid - 1);
        // 递归地构建右子树
        node.right = buildBST(nums, mid + 1, right);
        // 返回构建的二叉搜索树的根节点
        return node;
    }
}

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

可以用二叉搜索树的反向中序遍历(右-根-左)来解决。反向中序遍历二叉搜索树可以得到一个递减的序列。遍历时,用一个变量sum来存储已经遍历过的节点的值的总和,然后把这个总和赋值给当前节点。

class Solution {
    // 初始化 sum 为 0
    int sum = 0;

    public TreeNode convertBST(TreeNode root) {
        // 如果树为空,直接返回 null
        if (root == null) {
            return null;
        }
        // 首先遍历右子树
        convertBST(root.right);
        // 将当前节点的值加到 sum 中
        sum += root.val;
        // 将 sum 赋值给当前节点
        root.val = sum;
        // 然后遍历左子树
        convertBST(root.left);
        // 返回修改后的根节点
        return root;
    }
}

在这段代码中,变量sum用于存储已经遍历过的节点的值的总和。函数convertBST首先遍历右子树,然后处理根节点,最后遍历左子树,这样就实现了反向中序遍历。在处理根节点时,把sum赋值给当前节点,然后更新sum的值。最后返回修改后的根节点。

什么是累加树?

累加树是一种特殊的二叉树,其中每个节点的新值等于原树中大于或等于该节点值的所有值之和。

例如,给定一个如下的二叉搜索树:

  5
 / \
2   13

转换后,树的形状保持不变,只是节点值变为原来大于等于该节点值的所有节点值之和:

  18
 /  \
20   13

转换规则是这样的:根据二叉搜索树的特性,一个节点的值等于它的右子树的所有节点值之和加上它自己的值和它左子树中所有比它大的节点的值之和。所以,如果我们用逆中序遍历(右-根-左的顺序)来遍历这棵树,那么我们可以使用一个累加器(accumulator)来记录已经累加过的节点值,然后在遍历到一个新节点时,把累加器的值加到新节点的值中。这样,我们就可以在遍历的过程中完成节点值的转换,最终得到一个累加树。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值