leetcode--从二叉搜索树到更大和树

最新推荐文章于 2024-02-25 18:19:12 发布

原创最新推荐文章于 2024-02-25 18:19:12 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

leetcode之路专栏收录该内容

8 篇文章

订阅专栏

该博客介绍了如何将一个二叉搜索树转换为每个节点值等于原树中大于或等于其值的节点之和的树。通过反向中序遍历，利用已遍历节点的值之和更新当前节点值，从而实现转化。示例展示了具体操作过程，并指出此方法适用于最大值在叶子节点的情况。

给出二叉搜索树的根节点，该二叉树的节点值各不相同，修改二叉树，使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

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

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

示例：

输入：[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]

提示：

树中的节点数介于 1 和 100 之间。
每个节点的值介于 0 和 100 之间。
给定的树为二叉搜索树。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

主要思路：

首先，搜索二叉树是能保证任何非叶子节点的左子节点小于父节点，并且右子节点大于父节点的键值。

对于搜索二叉树而言，正向中序遍历是一个递增有序序列。反向中序遍历时一个递减的有序序列。

题意是需要将每个节点的值加上所有节点中大于或等于该节点的数值，所以对于最右边叶子节点，是不需要做任何操作的。以此类推，按照逆序遍历二叉搜索树中的节点，使用一个变量记录所有已经遍历过的节点的数值之和，当遇到下一个节点时，比这些节点的数值都小，加到后面的节点，更新该变量即可。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int sum = 0;
    public TreeNode bstToGst(TreeNode root) {
        post_dfs(root);
        return root ;
    }
    public void post_dfs(TreeNode root) {
        if(root == null){
            return ;
        }
        post_dfs(root.right);
        root.val += sum ;
        sum = root.val ;
        post_dfs(root.left);
        return ;
    }
}