代码随想录算法训练营day23|leetcode538/699/108

最新推荐文章于 2024-08-06 23:29:41 发布

yrrej0

最新推荐文章于 2024-08-06 23:29:41 发布

阅读量423

点赞数

文章标签： leetcode 算法职场和发展 python

本文链接：https://blog.csdn.net/weixin_53153619/article/details/127673524

版权

Leetcode538.把二叉搜索树转换为累加树

538. 把二叉搜索树转换为累加树 - 力扣（Leetcode）

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

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

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

思路：说实话看到这道题其实是很懵逼不知道二叉树的节点怎么进行累加，但是这是二叉搜索树！！！

Tips:只要看到二叉搜素树,统统的把他列成有序数组来看，然后思路就比较清晰了，这个题如果转换为有序数组来看的话这几个节点就是[0，1，2，3，4，5，6，7，8]然后根节点的累加值就是从右边往左去进行累加。

所以根节点0的新值就等于：8+.....0 = 36

1的新值就等于：8+....1 = 36

所以我们正常的中序遍历是左中右那么这道题就需要反中序遍历了。因为左边的值实际上的右边+中间节点得到的，所以遍历顺序就是右中左。

递归三部曲：

终止条件：节点为空即返回

传入参数:本题依然需要一个pre指针记录当前遍历节点root的前一个节点，这样才方便做累加。

所以需要定义一个新节点默认为0

单层循环逻辑：中间节点的值就等于当前节点加上上一个节点的值

代码：

def __init__(self):
        self.pre = TreeNode()


    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.traversal(root)
        return root

    def traversal(self,root):
        if not root:
            return None
        # 因为要遍历整棵树，所以递归函数不需要返回值
        # Base Case
        self.traversal(root.right)
        root.val+=self.pre.val
        self.pre = root 
        self.traversal(root.left)

Leetcode108将有序数组转为二叉搜索树

108. 将有序数组转换为二叉搜索树 - 力扣（Leetcode）

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

思路：

递归三部曲：终止条件:Nums=空就停止遍历

传入参数 nums

单层循环：找到root之后直接向左递归，向右递归。

这道题是比较简单的，和上面的题一样，看见是有序数组，也是二叉搜索树

首先确定我们的根节点，也就是左中右的顺序，那么根节点一定是我们这个有序数组中的中间位置

也就是mid = len(nums)//2，找到了root的下标，然后建立这个根节点。

之后分别向左递归和向右递归。返回根节点，那么这个有序数组就转化为了二叉搜索树。

代码：

def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:

        if not nums:return None
        mid = len(nums)//2
        root = TreeNode(nums[mid])
        root.left = self.sortedArrayToBST(nums[:mid])
        root.right = self.sortedArrayToBST(nums[mid+1:])
        return root

Leetcode669修建二叉搜索树

669. 修剪二叉搜索树 - 力扣（Leetcode）

思路：首先题目是给了一个界限，要求所有节点的值只能在这个界限当中，都是开区间。

如果直接一上来就判断当前节点的值如果大于或者小于边界，直接return none.这样子是不行的。

因为我们没有考虑如果子树还有左右孩子怎么办呢？

如果左右孩子的值是在边界中，那应该怎么办呢，我们是不是需要保留下来？

所以说我们需要对左右孩子也做一个处理：

如图：

这种就是一上来就写的代码，就没有考虑左右孩子的值是符合的。

那么看着像是需要改整棵树的结构了？？？

其实不需要的，在上图中我们发现节点0并不符合区间要求，那么将节点0的右孩子节点2 直接赋给节点3的左孩子就可以

递归三部曲：

终止条件：遇到空就返回

函数参数及返回值：我们使用递归需要考虑当前递归的结果需不需要返回值。

这里我们为什么需要返回值呢？因为是要遍历整棵树，做修改，其实不需要返回值也可以，我们也可以完成修剪（其实就是从二叉树中移除节点）的操作。

但是有返回值，更方便，可以通过递归函数的返回值来移除节点。

单层循环逻辑：如果root（当前节点）的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。

如果root(当前节点)的元素大于high的，那么应该递归左子树，并返回左子树符合条件的头结点。

所以需要返回值告诉上一层他新连接的孩子是谁。

接下来要将下一层处理完左子树的结果赋给root->left，处理完右子树的结果赋给root->right。

代码：

        def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if not root:
            return None
        if root.val<low:
            return self.trimBST(root.right,low,high)
            
        if root.val>high:
            return self.trimBST(root.left,low,high)
        
        root.left = self.trimBST(root.left,low,high)
        root.right = self.trimBST(root.right,low,high)
        
        return root

总结一下：做了这么多二叉树的题目了，几乎都是用的递归，很少去用迭代，在递归中经常用到的就是我到底是去搜索整条边呢，还是去搜索整棵树呢？到底需不需要返回值呢？

在上一篇的文章中我写到过搜索一条边的写法和搜索整棵树的写法

前提:如果需要返回值的话！