【LeetCode】把二叉搜索树转换为累加树 Java 题解

2021 年 12 月 16 日
本文字数：840 字
阅读完需：约 3 分钟

题目描述

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

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

节点的左子树仅包含键小于节点键的节点。节点的右子树仅包含键大于节点键的节点。左右子树也必须是二叉搜索树。

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

复制代码

思路分析

今天的算法每日一题是树相关的题目，题目要求使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。根据二叉搜索树的性质，中序遍历是升序，我们可以反向中序遍历，累加求出新的值。实现代码如下：

通过代码

/** * 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) {        if (root == null) {            return root;        }        convertBST(root.right);        sum += root.val;        root.val = sum;        convertBST(root.left);
        return root;
    }}