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

Posted 2022-11-05 风车菊

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

描述

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

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

节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。

来源：力扣（LeetCode）

思路

首先是一个二叉搜索树，可知中序遍历是一个有序的数组，先左后右是一个递增的，先右后左是一个递减的，再读题，我们要找出比当前节点大的所有节点和等于当前节点的值累加，那么我们采用先右后左的中序遍历，先找最大的，一个个计算，计算一个更新一个，哪些最大的永远是最新的值，再标记一个当前节点的前一个节点，就可很快计算出

代码

class Solution 
    TreeNode preNode = null;
    public TreeNode convertBST(TreeNode root) 
       demo(root);
       return root;
    
    public void demo(TreeNode root)
        if(root == null) return;
        demo(root.right);
        if(preNode != null)
            root.val = root.val + preNode.val;
         
        preNode = root;
        demo(root.left);