题目描述
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
解题思路
1、题意理解:
明确什么是二叉搜索树,任意结点的左子树比右子树小则为二叉搜索树
也就是说对于左子树来说,需要加上右子树的值和根节点的值
2、想法:
可以将二叉树排好序,如果我们希望每个结点只访问一次,排序的顺序是从大到小。如果使用中序遍历,我们可以得到从小到大的排序,中序遍历的顺序是left, root, right, 因此如果希望得到从大到小的排序,其结点的访问顺序是right,root, left。
代码
/**
* 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) {
convert(root);
return root;
}
public void convert(TreeNode cur){
if(cur == null){
return;
}
convert(cur.right);
cur.val = sum + cur.val;
sum = cur.val;
convert(cur.left);
}
}
参考:https://leetcode.com/problems/convert-bst-to-greater-tree/discuss/100506/Java-Recursive-O(n)-time