剑指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);
    }
}
相关推荐
林木辛20 分钟前
LeetCode热题 15.三数之和(双指针)
算法·leetcode·双指针
AndrewHZ27 分钟前
【3D算法技术】blender中,在曲面上如何进行贴图?
算法·3d·blender·贴图·三维建模·三维重建·pcg
Jared_devin39 分钟前
二叉树算法题—— [蓝桥杯 2019 省 AB] 完全二叉树的权值
数据结构·c++·算法·职场和发展·蓝桥杯
AI 嗯啦2 小时前
数据结构深度解析:二叉树的基本原理
数据结构·算法
和光同尘@3 小时前
66. 加一 (编程基础0到1)(Leetcode)
数据结构·人工智能·算法·leetcode·职场和发展
CHEN5_023 小时前
leetcode-hot100 11.盛水最多容器
java·算法·leetcode
songx_993 小时前
leetcode18(无重复字符的最长子串)
java·算法·leetcode
max5006004 小时前
实时多模态电力交易决策系统:设计与实现
图像处理·人工智能·深度学习·算法·音视频
其古寺4 小时前
贪心算法与动态规划:数学原理、实现与优化
算法·贪心算法·动态规划
rit84324994 小时前
基于灰狼算法(GWO)优化支持向量回归机(SVR)参数C和γ的实现
c语言·算法·回归