538.把二叉搜索树转换为累加树

给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如:

输入: 二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

使用递归:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


int sum=0;
struct TreeNode* convertBST(struct TreeNode* root){
    if(root==NULL)  return root;
    convertBST(root->right);
    root->val+=sum;
    sum=root->val;
    convertBST(root->left);
    return root;
}

可是在上述代码进行提交验证时,出现了一个小问题:

而第一个用例为:[5,2,13]  在运行完第一个用例后sum==20,所以进行第二个用例计算时,sum的初值为20。原因我猜测是:测试用例的程序在第一次测试时进行了函数外 int sum=0;的语句执行,但在后续函数测试时,直接进行了convertBST()的重复调用,没有执行函数外 int sum=0;的语句,导致sum初值发生错误;

所以,我将代码改为:

void sumBST(struct TreeNode* root,int * sum){
    if(root==NULL)  return root;
    sumBST(root->right,sum);
    root->val+=*sum;
    *sum=root->val;
    sumBST(root->left,sum);
}
struct TreeNode* convertBST(struct TreeNode* root){
    int sum=0;
    sumBST(root,&sum);
    return root;
}

在函数中调用函数,并在每次调用前对sum值赋值为0;这样在系统跑测试用例时就会每次都对sum进行初始化为0的操作。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值