538. 把二叉搜索树转换为累加树 题解

class Solution {

    // 先将右子树转换为累加树；并记录右子树的累加和rightSum;
    // 然后处理根节点，根节点的值 = 根节点值 + rightSum;
    // 然后转化左子树
    public TreeNode convertBST(TreeNode root) {
        convert(root, 0);
        return root;
    }
    
    public int convert(TreeNode root, int rightSum){
        if(null == root){
            return rightSum;
        }
        rightSum = convert(root.right, rightSum);
        root.val = root.val + rightSum;
        rightSum = root.val;
        rightSum = convert(root.left, rightSum);
        return rightSum;
    }
}

class Solution {
    
    // !!! 不太很优雅，但却让代码更容易理解!!!
    private int rightSum = 0;

    // 先将右子树转换为累加树；并记录右子树的累加和rightSum;
    // 然后处理根节点，根节点的值 = 根节点值 + rightSum;
    // 然后转化左子树
    public TreeNode convertBST(TreeNode root) {
        if(null == root){
            return root;
        }
        convertBST(root.right);
        root.val = root.val + rightSum;
        rightSum = root.val;
        convertBST(root.left);
        return root;
    }
}