Leetcode 538.二叉树转换为累加树

Time: 20190907
Type: Easy

题目描述

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

例如：

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

BST的中序遍历是一个递增序列，显然每个结点值是后续所有结点值之和。

按照逆中序遍历，过程中累加，并将累加的和添加到当前结点即可。用递归的思路比较好解。

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def __init__(self):
        self.num = 0
    def convertBST(self, root: TreeNode) -> TreeNode:
        # 中序遍历后的序列是递增序列
        if not root:
            return None
        
        self.convertBST(root.right)
        
        root.val += self.num
        
        self.num = root.val # 保存的是累加过后的数值
        
        self.convertBST(root.left)
        return root

END.