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.

样例

Given a binary search Tree `{5,2,3}`:

              5
            /   \
           2     13

Return the root of new tree

             18
            /   \
          20     13

解题思路:

题目要实现树上节点数值变为原数值加上所有比自己大的数值,形成一个比原来的树更大的树。用sum储存某些节数值的加和,根节点等于本身加右子树上的所有数值,左子树上的数值等于根节点的数值加右子树的数值再加本身数值,右子树的数值等于本身数值加子树上比自己大的数值。

代码实现:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root the root of binary tree
     * @return the new root
     */
    int sum=0;
    TreeNode* convertBST(TreeNode* root) {
        // Write your code here
        if(root==NULL) return NULL;
        else{
            convertBST(root->right);
            root->val+=sum;
            sum=root->val;
            convertBST(root->left);
            return root;
        }
    }
};

解题感悟:

这道题逻辑性需求比较强,一开始把convertBST(root->right)和root->val+=sum; sum=root->val;的顺序写反了,出现wrong answer。应该先访问右子树一直到右子树的叶子节点,储存叶子节点的数值,再对双亲节点操作同时储存加和,最后访问左子树对左子树进行操作。
            

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值