538.Convert BST to Greater Tree

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

Example:

Input: The root of a Binary Search Tree like this:
              5
            /   \
           2     13

Output: The root of a Greater Tree like this:
             18
            /   \
          20     13

 题意:

首先需要清楚BST,二叉搜索树的特征:

1. 每个节点都有一个作为搜索依据的关键码( key) , 所有节点的关键码互不相同。

2. 左子树上所有节点的关键码( key) 都小于根节点的关键码( key) 。

3. 右子树上所有节点的关键码( key) 都大于根节点的关键码( key) 。

4. 左右子树都是二叉搜索树。

然后题目要求变化这颗树为更大的树,即每个节点值加上所有大于他的节点值,成为新的值。然后返回这颗变大的树。

解法:由于是二叉搜索树,所以我们知道每棵树上他的所有右节点的值都要比他大,所以采用后序遍历,将右子树的所有节点和求出来然后与自身相加并重新赋值,就能得到更大的树。

class Solution {
public: int Sum = 0;
public:
    TreeNode* convertBST(TreeNode* root) {
        LrdSum(root);
        return root;
    }
    void LrdSum(TreeNode* root){
        if(root == NULL){
            return;
        }
        LrdSum(root->right);
        Sum+=root->val;
        root->val = Sum;
        LrdSum(root->left);
    }
};

总的来说知道二叉搜索树(BST)后,这题就没什么难度了。时间复杂度也为O(N);

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值