剑指Offer || 054.把二叉搜索树转换为累加树

题目

给定一个二叉搜索树,请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。

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

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

示例 1:

复制代码
输入:root=[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]

提示:

  • 树中的节点数介于 0104 之间。
  • 每个节点的值介于 -104104 之间。
  • 树中的所有值 互不相同
  • 给定的树为二叉搜索树。

注意:

LCR 054. 把二叉搜索树转换为累加树 - 力扣(LeetCode)

题解

思路一:两次dfs,第一次计算出sum,第二次中序遍历,将sum传给t.val,然后sum-=t.val

代码:

java 复制代码
class Solution {
	int sum=0;
    public TreeNode convertBST(TreeNode root) {
    	if(root==null) return null;
    	dfs(root);
    	dfs2(root);
    	return root;
    }
    
    public void dfs(TreeNode t) {
    	if(t.left!=null) dfs(t.left);
    	sum+=t.val;
    	if(t.right!=null) dfs(t.right);
    }
    public void dfs2(TreeNode t) {
    	if(t.left!=null) dfs2(t.left);
    	int tmp=t.val;
    	t.val=sum;
    	sum-=tmp;
    	if(t.right!=null) dfs2(t.right);
    }
}

思路二:中序遍历的倒序,每次遍历到TreeNode时sum+=t.val,然后t.val=sum

代码:

java 复制代码
class Solution {
	int sum=0;
    public TreeNode convertBST(TreeNode root) {
    	if(root==null) return null;
    	dfs(root);
    	return root;
    }
    
    public void dfs(TreeNode t) {
    	if(t.right!=null) dfs(t.right);
    	sum += t.val;
        t.val = sum;
    	if(t.left!=null) dfs(t.left);
    }
}
相关推荐
大怪v7 小时前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工9 小时前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农10 小时前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了10 小时前
AcWing学习——双指针算法
c++·算法
moonlifesudo11 小时前
322:零钱兑换(三种方法)
算法
NAGNIP1 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队1 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja1 天前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下1 天前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶1 天前
算法 --- 字符串
算法