leetcode_1038. 从二叉搜索树到更大和树

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

二、解题思路

中序遍历,记录右节点的值,然后递归返回让根节点加上该值更新根节点,接着左节点加上更新后的根节点的值,如此递归反复。

三、代码

 
# 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:
        cur = root
        self.num = 0

        def dfs(root):
            if root is not None:
                dfs(root.right)
                root.val = root.val + self.num
                self.num = root.val
                dfs(root.left)
                return root
            else:
                return None

        ans = dfs(cur)
        return ans





if __name__ == '__main__':
    a = TreeNode(5)
    a.left = TreeNode(2)
    a.right = TreeNode(13)


    s = Solution()
    ans = s.bstToGst(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值