LeetCode小算法记录（七十七）把二叉搜索树转换为累加树

最新推荐文章于 2024-09-20 21:48:23 发布

夏至微凉、

最新推荐文章于 2024-09-20 21:48:23 发布

阅读量160

点赞数

本文链接：https://blog.youkuaiyun.com/qq_31404603/article/details/105636520

版权

本文介绍了一种将二叉搜索树转换为累加树的方法，通过回溯算法，确保每个节点值等于原值加上所有大于其值的节点之和。示例展示了输入原始二叉搜索树和转换后的累加树。

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

给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如：

输入: 原始二叉搜索树:
5
/ \
2 13

输出: 转换为累加树:
18
/ \
20 13

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

package leetCodeTest;

import util.TreeNode;


public class 把二叉搜索树转换为累加树 {
    public static void main(String[] args) {
        TreeNode treeNode = new TreeNode(5);
        treeNode.left = new TreeNode(2);
        treeNode.right = new TreeNode(13);
        TreeNode treeNode1 = convertBST(treeNode);
        printTree(treeNode1);
    }

    /**
     * 回溯算法，考虑对一个子树的详细操作然后进行递归操作所有子树。
     */
    private static int sum = 0;
    public static TreeNode convertBST(TreeNode root) {
        if (root != null){
            convertBST(root.right);
            sum += root.val;
            root.val = sum;
            convertBST(root.left);
        }
        return root;
    }
    public static void printTree(TreeNode treeNode){
        if (treeNode == null) return;
        System.out.print(treeNode.val + " ");
        printTree(treeNode.left);
        printTree(treeNode.right);
    }
}