一、题目
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
注意:题目给定的是二叉***搜索***树,我就是因为没有仔细看题以为只是二叉树而陷入了思维误区。
二叉搜索树:非空条件下一定满足左子树结点值<根结点值<右子树结点值
思路:首先肯定是要遍历所有结点,根据搜索树的性质应该选择***中序遍历***(左根右)。又因为题目中所说的“每个节点要加上所有比他大的数”,也就是要加上此节点右侧的所有数。为了达到这个目的,需要把所有数按大小***倒序***排,那么每个结点都可以加上已遍历过的值得到最终结果。综上所述,即为中序反序遍历。
图解:
三、代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def __init__(self):
self.total = 0
def convertBST(self, root: TreeNode) -> TreeNode:
if root is not None:
self.convertBST(root.right)
self.total += root.val
root.val = self.total
self.convertBST(root.left)
return root