【LeetCode 538】把二叉搜索树转换为累加树(Python)

一、题目
给定一个二叉搜索树(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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值