跳转至

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

1、题目描述

给出二叉搜索树的根节点,该二叉树的节点值各不相同,修改二叉树,使每个节点 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]

提示:

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

2、解题思路

  • 先使用DFS得到总的和
  • 然后使用中序遍历,每一次先将和赋值,然后和减去之前的val
# 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:
        total = 0

        def dfs_sum(node):
            nonlocal total
            if not node:
                return
            dfs_sum(node.left)
            total += node.val
            dfs_sum(node.right)


        def dfs(node: TreeNode):
            nonlocal total
            if not node:
                return
            dfs(node.left)
            temp = node.val
            node.val = total
            total -= temp
            dfs(node.right)

        dfs_sum(root)
        dfs(root)
        return root