538. 把二叉搜索树转换为累加树
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
解题思路:对二叉搜索树采用右中左的顺序遍历树节点,得到的值就是从大到小排列的。使用一个变量记录节点值的累计和,并将该变量记录的累计和不断赋予给对应的节点。
Python3代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def convertBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
self.sum_node = 0
def inorder(node):
if not node:
return None
inorder(node.right)
self.sum_node += node.val
node.val = self.sum_node
inorder(node.left)
inorder(root)
return root