Leetcode 1038.从二叉搜索树到更大和树（Binary Search Tree to Greater Sum Tree）

Leetcode 1038.从二叉搜索树到更大和树

1 题目描述（Leetcode题目链接）

  给出二叉 搜索 树的根节点，该二叉树的节点值各不相同，修改二叉树，使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
  提醒一下，二叉搜索树满足下列约束条件：

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

提示：

• 树中的节点数介于 1 和 100 之间。
• 每个节点的值介于 0 和 100 之间。
• 给定的树为二叉搜索树。

2 题解

  本题与第538题完全相同。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def bstToGst(self, root: TreeNode) -> TreeNode:
        stack = []
        num = 0
        node = root
        while node or stack:
            while node:
                stack.append(node)
                node = node.right
            node = stack.pop()
            node.val += num
            num = node.val
            node = node.left
        return root