Leetcode 538 - Convert BST to Greater Tree

题意

给一个BST，改变其节点的值，将其节点的值加上所有比他大的节点的值

思路

首先注意是BST，其中序遍历的节点是递增序列。即假设中序遍历后的结果是$a_0, a_1, a_2, .... a_n$，我们有$a_0 < a_1 < a_2 < ... < a_n$。

那么对于$a_i$，其改变后的值应该是$a_i + a_{i+1} + a_{i + 2} + ... + a_n$，即我们只需要将$a_i$之后遍历的节点值累加起来即可。为了达到这个效果，我们只需要将中序遍历的顺序倒过来：$a_n, a_{n - 1}, ... , a_2, a_1, a_0$，对于$a_i$，只需要累加上其之前遍历过的节点值即可。

所以，我们只需要右，根，左的去遍历节点，并且遍历到根节点时，累加上之前遍历过的节点值即可。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int sum = 0;
    void dfs(TreeNode* root) {
        if (!root) return;
        dfs(root->right);
        sum += root->val;
        root->val = sum;
        dfs(root->left);
    }

    TreeNode* convertBST(TreeNode* root) {
        dfs(root);
        return root;
    }
};