-
Notifications
You must be signed in to change notification settings - Fork 40
/
Copy pathBinarySearchTreeToGreaterSumTree.js
51 lines (48 loc) · 1.53 KB
/
BinarySearchTreeToGreaterSumTree.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// Source : https://leetcode.com/problems/binary-search-tree-to-greater-sum-tree
// Author : Dean Shi
// Date : 2019-05-26
/***************************************************************************************
*
* Given the root of a binary search tree with distinct values, modify it so that every
* node has a new value equal to the sum of the values of the original tree that are
* greater than or equal to node.val.
*
* 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: [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]
*
* Note:
*
* The number of nodes in the tree is between 1 and 100.
* Each node will have value between 0 and 100.
* The given tree is a binary search tree.
*
***************************************************************************************/
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var bstToGst = function(root) {
gst(root)
return root
};
function gst(node, sum = 0) {
return node
? gst(node.left, node.val += gst(node.right, sum))
: sum
}