leetcode 538. 把二叉搜索树转换为累加树(python)

最新推荐文章于 2025-03-25 11:32:52 发布

原创最新推荐文章于 2025-03-25 11:32:52 发布 · 564 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#其他

收录于

博客专栏收录该内容

182 篇文章

订阅专栏

介绍一种算法，将二叉搜索树转换为累加树。转换后的每个节点值等于原值加上所有大于该值节点值之和。示例展示了输入输出的变化，并提供了Python实现代码。

本文参考自: 原文地址

给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如：

输入: 二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

# 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 __init__(self):
        self.lSum = 0
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None
        self.convertBST(root.right)
        self.lSum += root.val
        root.val = self.lSum
        self.convertBST(root.left)
        return root