题目:
538. 把二叉搜索树转换为累加树
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
输入: 原始二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
思路
搜索二叉树,左节点小于根节点,跟节点小于右节点。
所以按题目要求,我们需要从右侧开始遍历。
代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def convertBST(self, root: TreeNode) -> TreeNode:
self.sumVal = 0
node = root
def fun(node):
if node:
fun(node.right)
self.sumVal += node.val
node.val = self.sumVal
fun(node.left)
else:
return node
fun(node)
return root