leetcode 538. 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

尝试递归解法
每一层返回的是以该节点为根的子树的和，期间调整子树上所有节点的值。
这里还需要记录，上个节点传过来的值，也就是比该节点大的值得和（部分和，从父节点继承）
我的直观解法

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode convertBST(TreeNode root) {
        int tp = sumGreater(root, 0);
        return root;
    }

    public int sumGreater(TreeNode root, int pre){
        if(root==null) return 0;
        int right = pre;
        if(root.right!=null){
            right = sumGreater(root.right, pre);
        }
        root.val = root.val + right;
        int left = 0;
        if(root.left!=null){
            left = sumGreater(root.left, root.val);
        }
        return left==0?root.val:left;
    }
}

其中pre是记录的上个节点传来得值
先更新右子树，传入的是比根大的求和
再更新根，这是根的值是所有比跟大的和加上根
再更新左子树，这是传入的是父节点的值，因为父节点已更新好
这里left=0表示左子树为空或返回的值为0

这里值得注意的是，如果左子树不为空，那么左子树返回的结果就是以父节点为根的子树求和

更加巧妙的解法

public class Solution {

    int sum = 0;

    public TreeNode convertBST(TreeNode root) {
        convert(root);
        return root;
    }

    public void convert(TreeNode cur) {
        if (cur == null) return;
        convert(cur.right);
        cur.val += sum;
        sum = cur.val;
        convert(cur.left);
    }

}

这里的sum在不同时候有两个角色，在遍历左子树之前，是根更新后的值，也就是根和右子树的求和
在遍历完左子树后，为整个树的和。这里的思想就和上面说的如果左子树不为空，那么左子树返回的结果就是以父节点为根的子树求和一致。