题目描述:给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。![在这里插入图片描述](https://img-blog.csdnimg.cn/20200821173424353.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xfY19jX2M=,size_16,color_FFFFFF,t_70#pic_center)
解题思路:中序遍历右—中—左。右子树都是大于根节点本身的,所以右子树的累加和就是根节点需要累加的数值。 而左子树有些特殊。
根的左根,要累加根节点的数值。左根要累加它自身的右子树的数值。 左根右子树,因为小于根节点,所以计算左根右子树的时候还要把根节点的数值算上去,因为根节点大于它。左根的左子树要累加左跟的数值。
代码:
class Solution {
private:
int cur_sum;
public:
void travel(TreeNode*root){
if(!root) return;
if(root->right) travel(root->right);
root->val=(cur_sum+=root->val);
if(root->left) travel(root->left);
}
TreeNode* convertBST(TreeNode* root) {
travel(root);
return root;
}
};