538. 把二叉搜索树转换为累加树

最新推荐文章于 2022-10-17 15:31:42 发布

lizju

最新推荐文章于 2022-10-17 15:31:42 发布

阅读量61

点赞数

分类专栏： LeetCode每日一题 python3

本文链接：https://blog.csdn.net/weixin_42227482/article/details/108702025

版权

python3 同时被 2 个专栏收录

36 篇文章 1 订阅

订阅专栏

LeetCode每日一题

14 篇文章 0 订阅

订阅专栏

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

记录所有点的值排序
更新

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
import heapq
class Solution:
    def convertBST(self, root: TreeNode) -> TreeNode:
        A = []
        def dfs(root):
            if root:
                heapq.heappush(A, root.val)
                dfs(root.left)
                dfs(root.right)
        dfs(root)        
        prefix = [0]
        A = sorted(A)
        for x in A:
            prefix.append(x + prefix[-1])
            
        def modify(root):
            if root:
                i = bisect.bisect_right(A, root.val)
                root.val += prefix[-1] - prefix[i]
                modify(root.left)
                modify(root.right)
        modify(root)        
        return root