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

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

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

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

代码实现

后序实现

class Solution {
    private List<TreeNode> nodes = new ArrayList<>();

    public TreeNode convertBST(TreeNode root) {
        inorders(root);
        for (int i = 1; i < nodes.size(); i++) {
            nodes.get(i).val = nodes.get(i - 1).val + nodes.get(i).val;
        }
        return root;
    }

    public void inorders(TreeNode root) {
        if (root == null) {
            return;
        }
        inorders(root.right);
        nodes.add(root);
        inorders(root.left);
    }
}

递归优化

class Solution2 {
    private int sum = 0;

    public TreeNode convertBST(TreeNode root) {
        inorders(root);
        return root;
    }

    public void inorders(TreeNode root) {
        if (root == null) {
            return;
        }
        inorders(root.right);
        sum += root.val;
        root.val = sum;
        inorders(root.left);
    }
}