题目描述
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