leedcode-1038从二叉搜索树到更大的树

最新推荐文章于 2025-03-30 01:00:00 发布

原创最新推荐文章于 2025-03-30 01:00:00 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

leedcode刷题记录专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种算法，用于将二叉搜索树转换为更大的总和树，其中每个节点的新值等于原树中大于或等于其原始值的所有节点值之和。通过遍历和累积求和实现这一目标。

给出二叉搜索树的根节点，该二叉树的节点值各不相同，修改二叉树，使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。

示例：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

提示：

树中的节点数介于 1 和 100 之间。
每个节点的值介于 0 和 100 之间。
给定的树为二叉搜索树。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

15:47~ 16:07
思路:由于是排序好的二叉树,只用算出和之后每次便利时减去最小的给入即可:
如下是我代码:

# 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 sortkey(self, node,m_list):
        if node.left != None:
            self.sortkey(node.left,m_list)
        m_list.append(node.val)
        if node.right != None:
            self.sortkey(node.right,m_list)
    
    def resetkey(self,node):
        if node.left != None:
            self.resetkey(node.left)
        tmp = node.val
        node.val = self.m_sum
        self.m_sum -= tmp
        if node.right != None:
            self.resetkey(node.right)
            
    def bstToGst(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        m_list = []
        self.sortkey(root,m_list)
        self.m_sum = sum(m_list)
        self.resetkey(root)
        return root