给定一个二叉搜索树(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的操作。