题目:
给定一个二叉搜索树,请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
输入:root = [4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
分析:
通常,二叉搜索树的中序遍历按照节点的值从小到达按顺序遍历,也就是当遍历到某个节点时比该节点指小的节点都已经遍历过了,因此知道了所有比该节点的值小的所有节点的值之和sum,但题目要求把每个节点的值替换成大于或等于该节点的值的所有节点的值之和,因此需要先遍历下这个二叉搜索树,求出所有节点的值total,再用total减去sum即可,因此需要共遍历两次二叉搜索树,第一次只需遍历所有节点即可,第二次必须采用中序遍历。
能否做到只需遍历一次二叉搜索树就能完成所有操作,通常的中序遍历是先遍历左子树再遍历根节点,最后再遍历右子树,由于左子树节点的值较小,右子树节点的值较大,因此总体上是按照节点的值从小到大遍历,如果节点值按从大到小遍历,那么只需要改变中序的遍历顺序即可,先遍历右子树,再遍历根节点,最后再遍历左子树。利用变量sum来累加遍历过的节点的值,当遍历到一个节点时,值比它大的节点都已经遍历过了,因此sum就是所有节点大于或等于当前节点的值之和,因此用宿sum来替换当前节点值即可。
代码:
import java.util.Stack;
public class ConvertBST {
public TreeNode convertBST(TreeNode root){
Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
int sum = 0;
while (cur !=null && !stack.isEmpty()){
while (cur!=null){
stack.push(cur);
cur = cur.right;
}
cur = stack.pop();
sum += cur.val;
cur.val = sum;
cur = cur.left;
}
return root;
}
}