leetcode_538. 把二叉搜索树转换为累加树
【摘要】 目录
一、题目内容
二、解题思路
三、代码
一、题目内容
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 原始二叉搜索树: ...
目录
一、题目内容
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 原始二叉搜索树:
5
/ \
2 13输出: 转换为累加树:
18
/ \
20 13
二、解题思路
中序遍历,记录右节点的值,然后递归返回让根节点加上该值更新根节点,接着左节点加上更新后的根节点的值,如此递归反复。
三、代码
-
# Definition for a binary tree node.
-
class TreeNode:
-
def __init__(self, x):
-
self.val = x
-
self.left = None
-
self.right = None
-
-
class Solution:
-
def convertBST(self, root: TreeNode) -> TreeNode:
-
cur = root
-
self.num = 0
-
-
def dfs(root):
-
if root is not None:
-
dfs(root.right)
-
root.val = root.val + self.num
-
self.num = root.val
-
dfs(root.left)
-
return root
-
else:
-
return None
-
-
ans = dfs(cur)
-
return ans
-
-
-
-
-
-
if __name__ == '__main__':
-
a = TreeNode(5)
-
a.left = TreeNode(2)
-
a.right = TreeNode(13)
-
-
-
s = Solution()
-
ans = s.convertBST(a)
-
print(ans)
文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。
原文链接:nickhuang1996.blog.csdn.net/article/details/108702925
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)