题目描述
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
示例1:
输入: 二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题方法和思路
带*的表示参考其他人的实现
中序遍历实现
思路:
不难发现二叉搜索树使用中序遍历的话(先左子树后右子树),会形成一个升序的序列,因此考虑中序的先右子树后左子树遍历,则遍历顺序为降序,当前节点的累加树的值为这个遍历顺序下的和加上当前节点本来的值。
输入: 二叉搜索树:
5
/ \
2 13
则先右后左的中序遍历顺序为13->5->2,对应的累加树的要求为加上所有值比当前节点的值要大的节点。
因此对于13来说,没有比他大的值,因此保持不变。
对于5来说,值应该更改为5+13。
对于2来说,值应该更改为2+5+13。
对于先右后左的中序遍历顺序来说,使用一个temp变量保存之前遍历的和几个很方便的求出当前节点对应的累加树的值
复杂度:
和遍历一次树的复杂度一样
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) O(n) O(n)
核心代码:
# 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:
temp = 0
def InOrderTraverse(root):
nonlocal temp
if root:
InOrderTraverse(root.right)
root.val = temp + root.val
temp = root.val
InOrderTraverse(root.left)
InOrderTraverse(root)
return root