二叉搜索树 – 把二叉搜索树转换为累加树

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

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

• 节点的左子树仅包含键 小于 节点键的节点。
• 节点的右子树仅包含键 大于 节点键的节点。
• 左右子树也必须是二叉搜索树。

二叉树两部曲：

• 1 确立遍历顺序 – 右中左
• 2 确立递归返回值，由于不涉及建树，则为void

Notes：

• 1 由于每个节点的val需要加上上一个节点的值，故可以使用一个变量存储上一个节点的值

具体代码如下：

class Solution {
public:
    int pre = 0;
    void traversal(TreeNode* root){
        if (root==nullptr) return ;
        convertBST(root->right);
        root->val += pre;
        pre = root->val;
        convertBST(root->left);
    }

    TreeNode* convertBST(TreeNode* root) {
        // 1 确定遍历顺序 - 右中左 2 确定递归返回值 void
        if (root == nullptr) return nullptr;
        traversal(root);
        return root;
    }
};