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

题目来源：https://leetcode-cn.com/problems/convert-bst-to-greater-tree/
个人公众号：小猿君的算法笔记

题目描述

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

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

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

示例：
输入：[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]

思路分析

我们举一个简单的例子，对于下面这棵二叉树，根节点的值为1，按照题目的要求，更新后根节点的值将变成1 + 2 = 3，左节点的值为0，更新后，左节点的值将变为1 + 2 = 3，右节点是当前数中的最大值，因此更新后的值仍为2。

这是对二叉搜索树性质的一个应用，对于二叉搜索树，如果我们做一个中序遍历，那么我们将会得到一个递增的有序数列，如果我们把中序遍历稍作改动，先遍历右子树，那么它的顺序将会变成右子树->根节点->左子树，最终我们将得到一个递减的有序数列。

题目要求找到所有大于等于当前节点的值作为当前节点的新值。如果我们获得了这样一个递减的有序数列，对于每一个节点，它最终的值，就是这个节点在递减有序数列的值与之前节点值的总和。

代码描述

class Solution {
    private int presum = 0;
    public TreeNode convertBST(TreeNode root) {
        if (root == null) {
            return root;
        }
        convertBST(root.right);
        root.val += presum;
        presum = root.val;
        convertBST(root.left);
        return root;
    }
}