Convert BST to Greater Tree
1、原题
Given a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus sum of all keys greater than the original key in BST.
2、题目解析
题目原意是说,给我们一个二叉查找树,然后要求我们对二叉查找树上的每一个数字进行修改,其值为原值加上树种比其大的数。
思考:第一眼看到这题,第一个想法就是对每一个节点遍历一边树,找到比其大的数字。但是感觉有些繁琐了些。于是又仔细想了想题目。发现题目中明确提到了是二叉查找树,这样的数有一个特点,就是其节点左边的都比该节点的值要小,节点右边的都比该节点的值要大。于是我们可以根据其特征,修改我们的算法。算法思路如下:
1、初始化sum值为0;
2、访问节点的右节点,直到其右节点不存在。
3、设置该右节点的值current.val += sum;设置sum值,sum = current.val(当前节点值);
4、访问左节点,重复刚才过程。
具体原理就是因为二叉查找树种比一节点数要大的数都在其右边,我们通过递归的方法获取其所有右节点的累加和即可。
3、代码实现
public class Solution {
//初始化
private int sum = 0;
//递归寻找其大值之和
public void post(TreeNode current) {
if (current != null) {
post(current.right);
current.val += sum;
sum = current.val;
post(current.left);
}
}
public TreeNode convertBST(TreeNode root) {
post(root);
return root;
}
}