题目:
给定一棵二叉搜索树,返回一个更大的二叉搜索树。使得每一个节点的值都为之前该节点的值加上之前树中比该节点大的所有数之和。样例如下:
解题思路:
已知二叉搜索树的特性:对于每个节点来说,该节点的值必定大于该节点左子树的所有数,也必定小于该节点右子树的所有数。所以对该二叉搜索树进行后序遍历,并保存当前遍历过的所有节点之和,将其加到遍历到的节点上即可。
代码(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