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

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

说明

给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

示例

输入: 原始二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

题解思路

二叉搜索树是一棵空树,或者是具有下列性质的二叉树:

1、若它的左子树不空,则左子树上所有节点的值均小于它的根节点的值;
2、若它的右子树不空,则右子树上所有节点的值均大于它的根节点的值;
3、它的左、右子树也分别为二叉搜索树。
所以进行反向的中序遍历进行计算

代码实现

# 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
        """
        deque = collections.deque()
        def dfs(root):
            if root:
                dfs(root.right)
                deque.append(root.val)
                root.val = sum(deque)
                dfs(root.left)
        dfs(root)
        return root
©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页