写点什么

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

作者:HQ数字卡
  • 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;
}}
复制代码

总结

  • 上述算法的时间复杂度是 O(n), 空间复杂度是 O(n)

  • 坚持算法每日一题,加油!

发布于: 1 小时前阅读数: 5
用户头像

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

评论

发布
暂无评论
【LeetCode】把二叉搜索树转换为累加树Java题解