0
点赞
收藏
分享

微信扫一扫

Binary Search Tree to Greater Sum Tree

唯米天空 2022-03-30 阅读 67
leetcode

Given the root of 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 the sum of all keys greater than the original key in BST.

As a reminder, a binary search tree is a tree that satisfies these constraints:

  • The left subtree of a node contains only nodes with keys less than the node's key.
  • The right subtree of a node contains only nodes with keys greater than the node's key.
  • Both the left and right subtrees must also be binary search trees.

 

Example 1:

Input: root = [4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
Output: [30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

思路:reverse inorder traverse,也就是inorder: left, current, right;

现在需要right, current, left, 然后value进行累加。逆着写inorder就可以了。

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/

class Solution {
public TreeNode bstToGst(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
pushRight(root, stack);
TreeNode oldroot = root;

Integer preval = null;
while(!stack.isEmpty()) {
TreeNode node = stack.pop();
if(preval == null) {
preval = node.val;
} else {
node.val = node.val + preval;
preval = node.val;
}
if(node.left != null) {
pushRight(node.left, stack);
}
}
return oldroot;
}

public void pushRight(TreeNode node, Stack<TreeNode> stack) {
while(node != null) {
stack.push(node);
node = node.right;
}
}
}

 

举报

相关推荐

0 条评论