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

最新推荐文章于 2020-12-27 11:10:59 发布

原创最新推荐文章于 2020-12-27 11:10:59 发布 · 749 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode题解专栏收录该内容

39 篇文章

订阅专栏

本文介绍了一种算法，该算法将一棵普通的二叉搜索树转换为累加树，即每个节点的值变为该节点原始值加上其右侧所有节点的值。通过递归遍历树的右子树、根节点和左子树来实现这一转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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;
    }
}