问题描述:
给定一个二叉搜索树(Binary Search Tree
),把它转换成为累加树(Greater Tree
),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
问题分析:
特别注意 ,人家给的是二叉搜索树(Binary Search Tree
),也就是给的是二叉排序树,即父亲节点大于左孩子节点,小于右孩子节点。不是普通的二叉树了。
清楚了二叉排序树,要想求得一个节点的累加和, 很显然,好考虑两个传递方向,一是,它要累加自己所有的右孩子节点(可以理解从下向上传递),每个父节点,要把得到累加值传递给自己左孩子节点(可以理解为从上向下传递)。用递归实现即可。
Python3实现:
# 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
"""
def f(node, value):
if node == None: return None
f(node.right, value) # 递归 右孩子 的累加和
node.val += value[0] # 把有孩子的节点值,添加到本节点上
value[0] = node.val
f(node.left, value) # 把累加后的值,在传递给 左孩子
f(root, [0])
return root
声明: 总结学习,有问题或不当之处,可以批评指正哦,谢谢。