问题
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例子
思路
二叉搜索树的性质:左<跟<右
如何让每个节点的值=原来的节点值加上所有大于它的节点值之和?
即从大到小倒排序【通过中序的倒遍历,遍历每一个结点】,该节点的值为到目前为止的总和
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int res = 0;
public TreeNode convertBST(TreeNode root) {
if (root==null) return null;
root.right = convertBST(root.right);
root.val += res;
res = root.val;
root.left = convertBST(root.left);
return root;
}
}
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
res=0
# 中序的倒遍历,遍历每一个结点
def convertBST(self, root: TreeNode) -> TreeNode:
if root==None: return None
root.right = self.convertBST(root.right)
root.val += self.res
self.res = root.val
root.left = self.convertBST(root.left)
return root