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

最新推荐文章于 2022-05-02 09:45:00 发布

原创最新推荐文章于 2022-05-02 09:45:00 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #数据结构 #java

LeetCode 专栏收录该内容

78 篇文章

订阅专栏

这篇博客介绍了如何使用递归法将一个二叉搜索树转换为累加树，使得每个节点的值等于原树中大于或等于其值的所有节点之和。提供的Java代码实现中，通过先遍历右子树累加，然后更新当前节点值，最后遍历左子树，确保了在O(n)的时间复杂度内完成转换。

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

1.题目

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

【medium】

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。
注意：本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同

示例 1：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]

输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：

输入：root = [0,null,1]

输出：[1,null,1]

示例 3：

输入：root = [1,0,2]

输出：[3,3,2]

示例 4：

输入：root = [3,2,4,1]

输出：[7,9,4,10]

提示：

树中的节点数介于 0 和 $10^4$ 之间。

每个节点的值介于 $10^4$ 和 $10^4$ 之间。

树中的所有值互不相同。

给定的树为二叉搜索树。

2.解题

方法一：递归法

根据题意，先遍历右子树，将节点的值累加，并将累加后的值重新赋值给节点，最后再遍历左子树。

java：

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

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