求:
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 原始二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
注意:本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同
解:
思路:反向中序遍历,记录前面出现的所有节点的值的和,加入到当前节点。递归操作完成后,返回根节点的指针。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
void
convert(
struct
TreeNode* root,
int
* sum){
if
(root==NULL)
return
;
convert(root->right,sum);
*sum += root->val;
root->val = *sum;
convert(root->left,sum);
}
struct
TreeNode* convertBST(
struct
TreeNode* root){
if
(root==NULL)
return
root;
int
sum =
0
;
convert(root,&sum);
return
root;
}
来源:oschina
链接:https://my.oschina.net/u/4469818/blog/4284086