目录

一、题目内容

二、解题思路

三、代码

一、题目内容

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

文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。

原文链接：nickhuang1996.blog.csdn.net/article/details/108703486