题目来源:https://leetcode-cn.com/problems/convert-bst-to-greater-tree/
个人公众号:小猿君的算法笔记
题目描述
给出二叉搜索树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
- 节点的左子树仅包含键小于节点键的节点。
- 节点的右子树仅包含键大于节点键的节点。
- 左右子树也必须是二叉搜索树。
示例:
输入:[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
思路分析
我们举一个简单的例子,对于下面这棵二叉树,根节点的值为1,按照题目的要求,更新后根节点的值将变成1 + 2 = 3,左节点的值为0,更新后,左节点的值将变为1 + 2 = 3,右节点是当前数中的最大值,因此更新后的值仍为2。
这是对二叉搜索树性质的一个应用,对于二叉搜索树,如果我们做一个中序遍历,那么我们将会得到一个递增的有序数列,如果我们把中序遍历稍作改动,先遍历右子树,那么它的顺序将会变成右子树->根节点->左子树,最终我们将得到一个递减的有序数列。
题目要求找到所有大于等于当前节点的值作为当前节点的新值。如果我们获得了这样一个递减的有序数列,对于每一个节点,它最终的值,就是这个节点在递减有序数列的值与之前节点值的总和。
代码描述
class Solution {
private int presum = 0;
public TreeNode convertBST(TreeNode root) {
if (root == null) {
return root;
}
convertBST(root.right);
root.val += presum;
presum = root.val;
convertBST(root.left);
return root;
}
}