0
点赞
收藏
分享

微信扫一扫

【LeetCode538】把二叉搜索树转换为累加树(BST中序)

豆丁趣 2022-07-14 阅读 66


1.题目

【LeetCode538】把二叉搜索树转换为累加树(BST中序)_中序


【LeetCode538】把二叉搜索树转换为累加树(BST中序)_中序_02

2.思路

一定注意有BST的条件,BST的特性是中序遍历(左中右)得到从小到大的序列,而题目求的是大于等于当前结点的值替换原值——注意这里是大于等于!!所以就不是单纯的中序,而是逆中序(右中左)。
换句话说从题目的例子我们也能感性观察到左下方一坨的新值是比较大的,而右下方的新值是比较小的,而且我们的逆中序遍历,在逆中序遍历中,我们的全局遍历total在累加的过程依次赋值给当前的结点值,这个赋值过程也是从右下方到开始,再到左边迁移

3.代码

# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def convertBST(self, root: TreeNode) -> TreeNode:
def dfs(root: TreeNode):
nonlocal total # 用外面的total
if root:
dfs(root.right)
total += root.val
root.val = total
dfs(root.left)

total = 0
dfs(root) # 一定注意这是二叉搜索树,存储的值有规律!
return


举报

相关推荐

0 条评论