【力扣538. 把二叉搜索树转换为累加树】递归（python3）

最新推荐文章于 2024-08-05 16:02:23 发布

原创最新推荐文章于 2024-08-05 16:02:23 发布 · 152 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dfs #python #leetcode #算法 #二叉树

DFS-树专栏收录该内容

21 篇文章

订阅专栏

本文解析了如何通过递归实现LeetCode题目`Convert BST to Greater Tree`，通过逆中序遍历调整节点值，保持右侧节点大于左侧。关键在于维护全局最大值。

题目描述

https://leetcode-cn.com/problems/convert-bst-to-greater-tree/

思路题解

死活没想到怎么让右边的最大的节点获取到左边最小的节点，看了答案发现直接一个全局变量保存当前的最大值解决了。因为是逆中序遍历，顺序是右、中、左，因此遍历的值一定是单调递减的。

class Solution:
    def convertBST(self, root: TreeNode) -> TreeNode:
        def dfs(root: TreeNode):
            nonlocal total
            if root:
                dfs(root.right)
                total += root.val
                root.val = total
                dfs(root.left)
        
        total = 0
        dfs(root)
        return root