LeetCode-Python-538. 把二叉搜索树转换为累加树
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。例如:输入: 二叉搜索树:5/\213输出: 转换为累加树:18/...
·
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
第一种思路:
先得到中序遍历,然后得到整个树的结点值的和,
接着再遍历一次树,对每个节点加上 整个树的和 - 它自身和左子树之和。
这种方法比较慢。
class Solution(object):
def convertBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
#每个节点应该加上整个树的和 减去 它自身和左子树之和
if root is None:
return None
def inOrder(node):
if not node:
return []
return inOrder(node.left) + [node.val] + inOrder(node.right)
inorder = inOrder(root)
dp = [0 for _ in range(len(inorder))]
dp[0] = inorder[0]
treesum = sum(inorder)
for i in range(1, len(dp)):
dp[i] = inorder[i] + dp[i - 1]
def change(node):
if not node:
return
pos = inorder.index(node.val)
node.val += treesum - dp[pos]
change(node.left)
change(node.right)
change(root)
return root
第二种思路:
直接按照右中左的顺序中序遍历整棵树,
用一个变量s记录下需要加的值。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def convertBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
#右中左的遍历顺序
if not root:
return root
self.s = 0
def convert(node):
if not node:
return
convert(node.right)
node.val += self.s
self.s = node.val
convert(node.left)
convert(root)
return root
更多推荐

所有评论(0)