题目: 给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
示例:
思路:
1.根据题意可知输出结果应为:
13=13+(0)
5=5+(13+0)
2=2+(5+13+0)
2.由1等号左侧可知产生了降序序列,利用BST中序遍历(左->根->右)得到升序数组的特性,可以使用反中序遍历(右->根->左),然后进行遍历直接可得到一棵新树。
算法复杂度分析
1.时间复杂度O(N):因为需要将一棵树所有节点遍历且仅遍历一次。
2.空间复杂度O(N):因为递归会产生使用调用栈的情况。
Code-递归
class Solution {
int sum = 0; //记录当前初始值
public TreeNode convertBST(TreeNode root) {
convertMiddleOrderTravelser(TreeNode root); //调用辅助函数
return root;
}
public void convertMiddleOrderTravelser(TreeNode root) {
if (root == null) { //递归出口,空树终止条件
return;
}
convertMiddleOrderTravelser(root.right); //先访问遍历右子树
/*对根结点做出改变*/
sum += root.val;
root.val = sum;
convertMiddleOrderTravelser(root.left); //再访问遍历左子树
}
}