1038.binary_search_tree_to_greater_sum_tree.md

To Index

1038.Binary Search Tree to Greater Sum Tree

难度:Medium

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.

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.

If we want to get this greater sum tree, we should add the rightmost node, then add it's parent's node, and then, add it's parent's left child node.
So we can set a sum to record the total sum from the rightmost node to the leftmose node. Code is as follows:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
int sum=0;
public:
    TreeNode* bstToGst(TreeNode* root) {
        if(!root ) return root;
        if(!root->left && !root->right) 
        {
                    root->val +=sum;
                    sum=root->val;
                    return root;
    }
    if(root->right)
    {
        root->right = bstToGst(root->right);

    }
    root->val +=sum;
    sum =root->val;
    if(root->left)
    {
        root->left = bstToGst(root->left);
    }
        return root;
    }


};

执行用时 :4 ms, 在所有 C++ 提交中击败了83.28%的用户
内存消耗 :9.3 MB, 在所有 C++ 提交中击败了100.00%的用户