LCR 054. 把二叉搜索树转换为累加树(中等)(主站538\1038)

https://leetcode.cn/problems/w6cpku/
本题与主站 538 题相同： https://leetcode-cn.com/problems/convert-bst-to-greater-tree/
本题与主站 1038 题相同：https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/
难度：☆☆☆

题目：

给定一个二叉搜索树，请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。

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

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

示例：

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

代码：DFS

二叉搜索树的特点就是越左越小，越右越大，那就是说树的最右子节点是最大值，从该点出发网上走，用右中左的形式往上走，值会越来越小。根据题意，比当前节点大于或者等于的节点值肯定是在自己的右下层，那么将当前值换成右下层层层递加上来的和。右中左就是反逆序。
Python

class Solution:
    def convertBST(self, root: TreeNode) -> TreeNode:
        sums = 0
        def dfs(node):
            if not node:
                return
            dfs(node.right)  # 递归右子树
            nonlocal sums
            sums += node.val  # 叠加
            node.val = sums  # 将当前节点替换为叠加后的和
            dfs(node.left)  # 递归左子树
        dfs(root)
        return root

Java

class Solution {
    private int sums;

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

    private void dfs(TreeNode node) {
        if (node == null) {
            return;
        }
        dfs(node.right);
        sums += node.val;
        node.val = sums;
        dfs(node.left);
    }
}