从二叉搜索树到更大和树
- leetcode -1038 题 从二叉搜索树到更大和树
- 解题思路
- 代码演示
- 二叉树专题
leetcode -1038 题 从二叉搜索树到更大和树
原题链接:
https://leetcode.cn/problems/binary-search-tree-to-greater-sum-tree/
题目描述
给定一个二叉搜索树 root (BST),请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。
提醒一下, 二叉搜索树 满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
示例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]
提示:
树中的节点数在 [1, 100] 范围内。
0 <= Node.val <= 100
树中的所有值均 不重复
解题思路
搜索二叉树的特性,左子树小于头节点,头节点小于右子树
左 < 头 < 右。
中序遍历顺序是: 左 头 右,
因此中序遍历是有序的,
但题中要求的是,大于节点值的累加和,
如果从左子树遍历,到左子树时,我们无法去累加后序节点的值
因此我们需要对中序遍历进行逆序
先遍历右子树,
然后依次向左遍历,左边都小于右边,就可以累加了
代码演示
/**
* 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 bstToGst(TreeNode root) {
if(root == null){
return null;
}
in(root);
return root;
}
/**
* 中序遍历的逆序 右 头 左
*/
public void in(TreeNode root){
if(root == null){
return;
}
//先遍历右树
in(root.right);
//中序累加和赋值给节点
sum += root.val;
root.val = sum;
//再遍历左树
in(root.left);
}
}
二叉树专题
根据前序和后序遍历构造二叉树
最大二叉树
从中序与后序遍历序列构造二叉树
二叉树的序列化和反序列化
leetcode 二叉树展开为链表