一、题目
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
二、思路——递归
累加树,是将每个节点的值改为原来的节点值加上所有大于它的节点值之和。而二叉搜索树的中序遍历的结果是从小到大排列的,比如,给定一个二叉搜索树如下:
这棵树中序遍历的结果为
2,3,4,5,6,7,8
由中序遍历结果我们很容易看到,一个节点的右边都是大于它的节点。若将中序遍历的结果反序,则得到
8,7,6,5,4,3,2
也就是反序中序遍历(右子树—>根节点—>左子树)。此时,对于每一个节点来说,大于它的节点在它的左边。因此,使用反序中序遍历时,就可以在遍历每个节点之前,先遍历所有大于它的节点。将这些节点记录并累加至当前节点,就可完成累加树。
由 8,7,6,5,4,3,2 得到累加树的过程如下:
- 8,7,6,5,4,3,2
- 8,15(8+7),6,5,4,3,2
- 8,15,21(15+6),5,4,3,2
- 8,15,21,26(21+5),4,3,2
- 8,15,21,26,30(26+4),3,2
- 8,15,21,26,30,33(30+3),2
- 8,15,21,26,30,33,35(33+2)
由原二叉搜索树得到的累加树为
三、代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int sum = 0;
public TreeNode convertBST(TreeNode root) {
if(root != null){
convertBST(root.right);
sum += root.val;
root.val = sum;
convertBST(root.left);
}
return root;
}
}