LeetCode:538. Convert BST to Greater Tree - Python

538. 把二叉搜索树转换为累加树

问题描述:

给定一个二叉搜索树(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

声明: 总结学习,有问题或不当之处,可以批评指正哦,谢谢。

题目链接

猜你喜欢

转载自blog.csdn.net/XX_123_1_RJ/article/details/87166923