【leetcode系列】【算法】【简单】把二叉搜索树转换为累加树

冰临天下

于 2020-04-16 22:30:08 发布

阅读量151

点赞数

分类专栏： leetcode系列

本文链接：https://blog.youkuaiyun.com/songyuwen0808/article/details/105568826

版权

leetcode系列专栏收录该内容

139 篇文章

订阅专栏

本文介绍了一种将二叉搜索树（BST）转换为大于树的方法，通过修改中序遍历的方式，从右子树开始遍历，利用栈记录节点，更新节点值为比其大的所有节点值之和。此方法巧妙地解决了LeetCode上的一道题目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

题目链接： https://leetcode-cn.com/problems/convert-bst-to-greater-tree/

解题思路：

类似中序遍历的栈的使用方式，参考：二叉树的中序遍历，只不过把遍历方式修改为根 -> 右 -> 左

设置一个变量add_num来记录比当前数字大的所有数字的和

代码实现：

# 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:
        if not root:
            return root
        
        add_num = 0
        stack = [root]
        while stack:
            node = stack[-1]
            if node.right:
                stack.append(node.right)
                continue
                
            while stack:
                node = stack[-1]
                node.val += add_num
                add_num = node.val
                stack.pop()
                if node.left:
                    stack.append(node.left)
                    break
                    
        return root