Leetcode.0538 | 把二叉树转化为累加树

题目

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（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]

输入：root = [0,null,1]
输出：[1,null,1]

输入：root = [1,0,2]
输出：[3,3,2]

输入：root = [3,2,4,1]
输出：[7,9,4,10]

解决办法

观察示例中的树，计算树中各个节点值之间的关系。从中可以发现，对数组进行右子树->根节点->左子树的遍历，然后将遍历结果从前到后进行累加，即可得到结果。

比如，示例1中，对数组进行右子树->根节点->左子树的遍历，结果如下所示：

[8 7 6 5 4 3 2 1 0]

然后在从前往后累加，形成：

[8 15 21 26 30 33 35 36 36]

所以，要解决这个问题，就需要进行右子树->根节点->左子树的遍历，然后按照顺序累加。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int sum = 0; //记录之前节点的累加和
    public TreeNode convertBST(TreeNode root) {
        if(root != null){
            convertBST(root.right);
            root.val = sum + root.val;
            sum = root.val;
            convertBST(root.left);
            return root;
        }
        return null;
    }
}