538. 把二叉搜索树转换为累加树
题目
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
- 节点的左子树仅包含键 小于 节点键的节点。
- 节点的右子树仅包含键 大于 节点键的节点。
- 左右子树也必须是二叉搜索树。
注意:本题和 1038 相同


提示:
树中的节点数介于 0 和 10⁴之间。
每个节点的值介于 -10⁴和 10⁴之间。
树中的所有值 互不相同 。
给定的树为二叉搜索树。
解题思路
这题首先确立思路。是要用中序遍历,但是中序遍历是把节点的值按照从小到大的顺序排列。
我们题目要求的确实大于等于当前节点的值的和。即我们要想到累加和。也就是要把节点的值按照从大到小排列(反中序遍历:先递归右子树,再递归左子树)。
然后每次遍历到root.val的时候,就加上大于root.val的值和它本身,但是因为用了反中序遍历,也就是加上root.val前面的值和它本身,这一看就是一个累加,即定义一个sum来保存累加的和。
要注意,sum不能定义在方法内,这样每次递归都会清零,要定义成全局的
代码
class Solution {
int sum=0;//注意这个sum不能放到方法里面,不然每次递归都会把sum归零,让前面的累加和失效。
public TreeNode convertBST(TreeNode root) {
if(root==null){
return root;
}
convertBST(root.right);
sum+=root.val;//用来保存累加和
root.val=sum;//把root的值更新为累计和
convertBST(root.left);
return root;
}
}