代码随想录算法训练营第二十五天 | 669. 修剪二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树、总结篇

目录

669.修剪二叉搜索树

思路

代码

思路

代码

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

思路

代码

总结篇 

---

669.修剪二叉搜索树

这道题目比较难，比 添加增加和删除节点难的多，建议先看视频理解。

题目链接/文章讲解： 代码随想录

视频讲解： 你修剪的方式不对，我来给你纠正一下！| LeetCode：669. 修剪二叉搜索树_哔哩哔哩_bilibili

思路

        这道题没有想象的那么简单但多思考一下还是知道什么原理的，（虽然我想了半天都没有想出来。。。）

        递归三部曲：

        1、确定递归函数的参数和返回值：每一次递归结束后都是返回节点。

        2、确定终止条件：遇到空节点就返回就可以了。

        3、确定单层递归的逻辑：如果当前节点小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。左子树同理。

代码

class Solution:
    def trimBST(self, root: TreeNode, low: int, high: int) -> TreeNode:
        if root is None:
            return None
        if root.val < low:
            # 寻找符合区间 [low, high] 的节点
            return self.trimBST(root.right, low, high)
        if root.val > high:
            # 寻找符合区间 [low, high] 的节点
            return self.trimBST(root.left, low, high)
        root.left = self.trimBST(root.left, low, high)  # root.left 接入符合条件的左孩子
        root.right = self.trimBST(root.right, low, high)  # root.right 接入符合条件的右孩子
        return root

---

 108.将有序数组转换为二叉搜索树  

本题就简单一些，可以尝试先自己做做。

题目链接：

代码随想录

视频讲解：构造平衡二叉搜索树！| LeetCode：108.将有序数组转换为二叉搜索树_哔哩哔哩_bilibili

思路

        别骂了别骂了，这道题Carl说是比较简单的，结果一开始我还搁那懵逼，甚至想着从第一个开始遍历（每日崩溃1/1），后面还是看了Carl哥的一行题解才想起是什么原理的。

        本质就是寻找分割点，分割点作为当前节点，然后递归左区间和右区间。

        直接从中间那个点开始（python直接整除2），然后划分为左边和右边，一直递归构建左右子树，就这么简单。

代码

from typing import List, Optional


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        if len(nums) == 0:
            return

        if len(nums) == 1:
            return TreeNode(nums[0])

        middle = len(nums) // 2
        root = TreeNode(nums[middle])

        nums_left = nums[:middle]
        nums_right = nums[middle + 1:]

        root.left = self.sortedArrayToBST(nums_left)
        root.right = self.sortedArrayToBST(nums_right)

        return root

---

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

本题也不难，在 求二叉搜索树的最小绝对差 和 众数 那两道题目 都讲过了 双指针法，思路是一样的。

题目链接：

代码随想录

视频讲解：普大喜奔！二叉树章节已全部更完啦！| LeetCode：538.把二叉搜索树转换为累加树_哔哩哔哩_bilibili

思路

        这道题还是用到了pre指针，就是我对pre指针的使用还是不太熟练 ，这里我甚至没想到一开始pre指针可以设置成0，这样就不用考虑初始pre指针没有的问题了。采用右中左的方式进行遍历。

代码

from typing import Optional


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution:
    def __init__(self):
        self.pre = 0  # 直接初始赋值为0

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

    def traversal(self, cur):
        if not cur:
            return None

        self.traversal(cur.right)  # 右

        cur.val += self.pre  # 中
        self.pre = cur.val

        self.traversal(cur.left)  # 左

        return cur

---

总结篇 

代码随想录

回顾了所有的二叉树题目，文末还有一张思维导图，记录了所有二叉树的解题策略。