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

最新推荐文章于 2025-12-17 19:02:04 发布

原创最新推荐文章于 2025-12-17 19:02:04 发布 · 121 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode #算法 #数据结构 #深度优先

二叉树专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种高效的方法来将二叉搜索树转换为累加树，通过两种不同的实现方式，一种是使用列表存储节点后进行遍历更新，另一种是采用反向中序遍历的方式直接修改节点值。

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

class Solution {
    List<TreeNode> arr;
    public TreeNode convertBST(TreeNode root) {
        arr = new ArrayList<>();
        dfs(root);
        for(int i = arr.size() - 2;i >= 0;i--){
            arr.get(i).val += arr.get(i + 1).val;
        }
        return root;
    }
    public void dfs(TreeNode root){
        if(root == null) return;
        dfs(root.left);
        arr.add(root);
        dfs(root.right);
    }
}

这真的是中等题嘛，秒过。

看了一下强中自有强中手。。。

class Solution {
    public TreeNode convertBST(TreeNode root) {
        dfs(root, 0);
        return root;
    }

    private int dfs(TreeNode root, int sum){
        //sum为该节点右边的所有节点的值的总和（所有比该节点大的节点）
        if(root == null){
            return sum;
        }
        //累加操作
        root.val += dfs(root.right, sum);
        //将累加后的结果传递给左边节点
        return dfs(root.left, root.val);
    }
}

利用了反向中序遍历，这样就是从大到小遍历，当前节点的值就是累加它自己和它的右子树的和。