题目来源: 538.把二叉搜索树转换为累加树
题目: 给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
注意:本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同
思路:反向中序遍历 累加树的形式需要每个结点的新值大于或等于节点值之和。鉴于二叉搜索树是一个有序的中序遍历,从小到大排列的数组,故,采用反向中序遍历的思想,得到一个从大到小的数组。每到一个新的结点时,结点新的值等于新的结点值加上之前结点的累加和,即可实现每个结点的新值大于或等于结点之和。
python代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def convertBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
self.total = 0
def dps(root):
if not root:
return
dps(root.right)
self.total += root.val
root.val = self.total
dps(root.left)
dps(root)
return root