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.
Example:
Input: The root of a Binary Search Tree like this: 5 / \ 2 13 Output: The root of a Greater Tree like this: 18 / \ 20 13
题意:
首先给定一个二叉排序树,即父节点的左孩子的值比它小,右孩子的值比它大。对每一个节点,需要加上所有比它大的节点的值,然后重新输出这棵二叉树。
题解:
此题有非常巧妙的方法,因为对于二叉排序树而言,对其进行中序遍历,可以得到一串从小到大排列的数,但是此题就巧妙在,按照先遍历右节点,然后是父亲节点,再然后是左节点的方式遍历(注意:这种方式不是后续遍历,后续遍历是右左根),因为右边的节点普遍是最大的,所以先遍历右节点,然后是其父亲节点;那么此刻就需要将父亲节点的值加上其右孩子的值,并将按照这种方式得到的新的父亲节点的值保存在一个变量sum中,作为需要给父亲节点的左孩子进行操作时的sum值。这里特别需要注意的是,在java中,对函数的参数传递是按照传引用而不是传值,所以如果在参数中设置了传值,那么返回的结果反而是不对的,并不会将已经计算得到的结果返回给上层,所以这里的代码上,我将sum放在外部变量来处理,这样可以保证递归调用函数的参数中没有这个sum值。此题用到的还是树的遍历,但是非常巧妙,灵活一变,当然对于递归的理解和调用也是要求比较高。虽是easy的题目,但是蕴含的知识点和想法一点不比medium和hard简单,需要多归纳这种类型的题目。
public class convertBST
{
int sum = 0; //这里将sum放在外部,作为外部变量来处理
public TreeNode convertBST(TreeNode root)
{
if(root == null)
return null;
convertBST(root.right); //先右子树
sum += root.val;
root.val = sum;
convertBST(root.left); //再左子树
return root;
}
}