LeetCode--538. Convert BST to Greater Tree(将二叉搜索树转为更大的二叉搜索树)

题目:
给定一棵二叉搜索树,返回一个更大的二叉搜索树。使得每一个节点的值都为之前该节点的值加上之前树中比该节点大的所有数之和。样例如下:


解题思路:
已知二叉搜索树的特性:对于每个节点来说,该节点的值必定大于该节点左子树的所有数,也必定小于该节点右子树的所有数。所以对该二叉搜索树进行后序遍历,并保存当前遍历过的所有节点之和,将其加到遍历到的节点上即可。
代码(Python):
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        def ReadTree(root,temp_sum):
            if root==None:
                return temp_sum
            if root!=None:
                temp_sum = ReadTree(root.right,temp_sum)
                root.val = root.val+temp_sum
                temp_sum = root.val
                temp_sum = ReadTree(root.left,temp_sum)
                return temp_sum
        
        ReadTree(root,0)
        return root
            


猜你喜欢

转载自blog.csdn.net/xiaoxiaoley/article/details/79975769