题目:
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node
的新值等于原树中大于或等于 node.val
的值之和。
思路:
二叉搜索树的中序遍历为递增序列,该题结果是把这个递增序列的每一个值转换为该值到数组最后一个值的累加。如果从后往前加会更加方便,不用每次都扫一遍数组。这个思想转换到遍历二叉搜索树就等价于倒序的中序遍历,即右中左。
代码:
private int pre=0;//记录前一个节点的值
public TreeNode convertBST(TreeNode root) {
traversal(root);
return root;
}
public void traversal(TreeNode root){
if(root==null) return;
traversal(root.right);
root.val=root.val+pre;
pre=root.val;
traversal(root.left);
}