​LeetCode刷题实战538:把二叉搜索树转换为累加树

共 2174字,需浏览 5分钟

 ·

2022-03-02 20:58

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

今天和大家聊的问题叫做 把二叉搜索树转换为累加树,我们先来看题面:
https://leetcode-cn.com/problems/convert-bst-to-greater-tree/

Given the root of a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus the sum of all keys greater than the original key in BST.


As a reminder, a binary search tree is a tree that satisfies these constraints:

1.The left subtree of a node contains only nodes with keys less than the node's key.

2.The right subtree of a node contains only nodes with keys greater than the node's key.

3.Both the left and right subtrees must also be binary search trees.


给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
  • 节点的左子树仅包含键 小于 节点键的节点。

  • 节点的右子树仅包含键 大于 节点键的节点。

  • 左右子树也必须是二叉搜索树。


示例                         


解题


思路:多写两层搜索树就会很明显的发现,所求的每个节点的值就是该节点的右子树所有节点的和加上自身的节点值,所以采用中序遍历,从右子树开始遍历即可,设置一个全局变量sum用于记录节点值的和。



/**
 * 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){
            convertBST(root.right);
            sum += root.val;
            root.val = sum;
            convertBST(root.left);
 
        }
        return root;
    }
}


好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。

上期推文:
LeetCode1-520题汇总,希望对你有点帮助!
LeetCode刷题实战521:最长特殊序列 Ⅰ
LeetCode刷题实战522:最长特殊序列 II
LeetCode刷题实战523:连续的子数组和
LeetCode刷题实战524:通过删除字母匹配到字典里最长单词
LeetCode刷题实战525:连续数组
LeetCode刷题实战526:优美的排列
LeetCode刷题实战527:单词缩写
LeetCode刷题实战528:按权重随机选择
LeetCode刷题实战529:扫雷游戏
LeetCode刷题实战530:二叉搜索树的最小绝对差
LeetCode刷题实战531:孤独像素 I
LeetCode刷题实战532:数组中的K-diff数对
LeetCode刷题实战533:孤独像素 II
LeetCode刷题实战534:游戏玩法分析 III
LeetCode刷题实战535:TinyURL 的加密与解密
LeetCode刷题实战536:从字符串生成二叉树
LeetCode刷题实战537:复数乘法

浏览 28
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报