剑指offer54:所有大于或等于节点的指之和

题目:
给定一个二叉搜索树,请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
在这里插入图片描述
输入:root = [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,但题目要求把每个节点的值替换成大于或等于该节点的值的所有节点的值之和,因此需要先遍历下这个二叉搜索树,求出所有节点的值total,再用total减去sum即可,因此需要共遍历两次二叉搜索树,第一次只需遍历所有节点即可,第二次必须采用中序遍历。
能否做到只需遍历一次二叉搜索树就能完成所有操作,通常的中序遍历是先遍历左子树再遍历根节点,最后再遍历右子树,由于左子树节点的值较小,右子树节点的值较大,因此总体上是按照节点的值从小到大遍历,如果节点值按从大到小遍历,那么只需要改变中序的遍历顺序即可,先遍历右子树,再遍历根节点,最后再遍历左子树。利用变量sum来累加遍历过的节点的值,当遍历到一个节点时,值比它大的节点都已经遍历过了,因此sum就是所有节点大于或等于当前节点的值之和,因此用宿sum来替换当前节点值即可。
代码:

import java.util.Stack;

public class ConvertBST {
    public TreeNode convertBST(TreeNode root){
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        int sum = 0;
        while (cur !=null && !stack.isEmpty()){
            while (cur!=null){
                stack.push(cur);
                cur = cur.right;
            }
            cur = stack.pop();
            sum += cur.val;
            cur.val = sum;
            cur = cur.left;
        }
        return root;
    }
}

在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

龙崎流河

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

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

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

打赏作者

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

抵扣说明:

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

余额充值