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

给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如:

输入: 二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

思路:

本题使用了类似中序遍历的迭代法,关于中序遍历的迭代法请看:二叉树的中序遍历

中序遍历的顺序:左子节点、根节点、右子节点;

本题是二叉搜索树,右子树所有节点的值都大于根节点,所以遍历的顺序应该是:右子节点、根节点、左子节点

设置变量curr=0,表示当前的累加值

迭代法如下:令p = root,当p不为空或栈中还有元素时进行以下循环

1、当p不为空时,将p的右子节点加入栈,并让p=p.right

2、出栈顶元素,让这个元素的值加上累加值,更新累加值为此元素的值,并让p=p.left

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var convertBST = function(root) {
  var stack = [],
      curr = 0,
      p = root;
  while (p || stack.length) {
    while (p) {
      stack.push(p)
      p = p.right
    }
    p = stack.pop()
    p.val += curr
    curr = p.val
    p = p.left
  }
  return root
};

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

进击的桐人

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值