538. Convert BST to Greater Tree*

https://leetcode.com/problems/convert-bst-to-greater-tree/

题目描述

Given a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus sum of all keys greater than the original key in BST.

Example:

Input: The root of a Binary Search Tree like this:
              5
            /   \
           2     13

Output: The root of a Greater Tree like this:
             18
            /   \
          20     13

C++ 实现 1

中序遍历, 使用一个 “global” 的变量 sum 来存储目前已经访问的节点值的和.

class Solution {
private:
    int sum = 0;
public:
    // 使用中序遍历, 但是先访问右子树, 再访问根节点, 最后访问
    // 左子树, 这样的话, 元素是从大到小排列, 使用 sum 记录访问
    // 到当前元素时, 前面已访问过的元素的和是多少, 访问完当前
    // 根节点的元素后, 不要忘记更新 sum 的值.
    TreeNode* convertBST(TreeNode* root) {
        if (!root)
            return nullptr;
        convertBST(root->right);
        root->val += sum;
        sum = root->val;
        convertBST(root->left);
        return root;
    }
};