首页 > 其他 > 详细

LeetCode 1038.从二叉搜索树到更大和树

时间:2020-03-11 22:44:56      阅读:86      评论:0      收藏:0      [点我收藏+]

??博客原文 :《LeetCode 1038.从二叉搜索树到更大和树 - JavaScript》

给出二叉搜索树的根节点,该二叉树的节点值各不相同,修改二叉树,使每个节点 node 的新值等于 原树中大于或等于 node.val 的值之和

解法:改造中序遍历

根绝二叉搜索树的性质,中序遍历就是按照从小到大的顺序遍历节点。假设中序遍历结果是:1、2、3、4。那么根据题目要求,每个节点的值更新如下:

  • 4 => 4
  • 3 => 4 + 3
  • 2 => 4 + 3 + 2
  • 1 => 4 + 3 + 2 + 1

假如先遍历右节点,再遍历当前节点,最后遍历左节点。因为之前遍历的节点都大于当前节点,所以当前节点的新值就等于 之前遍历的节点的和 + 节点自身的值

代码实现如下:

// ac地址:https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/
// 原文地址:https://xxoo521.com/2020-03-06-binary-search-tree-to-greater-sum-tree/

/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var bstToGst = function(root) {
    let sum = 0; // 之前遍历的所有节点的和
    travse(root);
    return root;

    function travse(node) {
        node.right && travse(node.right);

        node.val += sum;
        sum = node.val;

        node.left && travse(node.left);
    }
};

更多资料

若有错误,欢迎指正。若对您有帮助,请给个「关注+点赞」,您的支持是我更新的动力 ??

LeetCode 1038.从二叉搜索树到更大和树

原文:https://www.cnblogs.com/geyouneihan/p/12465125.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!