1038：从二叉搜索树到更大和树（1374 分）

力扣第 135 场周赛第 2 题

题目

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

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

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

示例 1：

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

示例 2：

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

提示：

• 树中的节点数在 [1, 100] 范围内。
• 0 <= Node.val <= 100
• 树中的所有值均 不重复 。

注意：该题目与 538: https://leetcode-cn.com/problems/convert-bst-to-greater-tree/ 相同

分析

和 0538 相同。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

def bstToGst(self, root: TreeNode) -> TreeNode:
    stack, pre = [(root, 0)], 0
    while stack:
        node, flag = stack.pop()
        if flag:
            node.val += pre
            pre = node.val
        elif node:
            stack.extend([(node.left, 0), (node, 1), (node.right, 0)])
    return root

48 ms