算法记录 | Day23 二叉树

最新推荐文章于 2024-06-21 17:14:10 发布

是菜鸡小小陈啊

最新推荐文章于 2024-06-21 17:14:10 发布

阅读量23

点赞数

分类专栏： leetcode 文章标签：算法数据结构

本文链接：https://blog.csdn.net/m0_37634594/article/details/129980113

版权

leetcode 专栏收录该内容

52 篇文章 4 订阅

订阅专栏

669. 修剪二叉搜索树

递归

思路：

1.确定递归参数和返回值：输入root、low、high

2.终止条件：修剪的操作并不是在终止条件上进行的，所以就是遇到空节点返回就可以了

3.单层逻辑：

如果root.va<low，那么应该递归右子树，并返回右子树符合条件的头结点
如果root.val>high，那么应该递归左子树，并返回左子树符合条件的头结点。
下一层处理完左子树的结果赋给root->left，处理完右子树的结果赋给root->right
返回root

class Solution:
    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)
        if low <= root.val <= high:
            root.left = self.trimBST(root.left, low, high)
            root.right = self.trimBST(root.right, low, high)
            return root

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

递归

思路：

1.确定递归参数和返回值：输入数组，左下标left，右下标right，用递归函数的返回值来构造中节点的左右孩子

2.终止条件：左闭右闭的区间，所以当区间 left > right的时候，就是空节点了返回

3.单层逻辑：

首先取数组中间元素的位置
接着划分区间，root的左孩子接住下一层左区间的构造节点，右孩子接住下一层右区间构造的节点
返回root

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

    def traversal(self, nums: List[int], left: int, right: int) -> TreeNode:
        if left > right:
            return None

        mid = left + (right - left) // 2
        mid_root = TreeNode(nums[mid])
        mid_root.left = self.traversal(nums, left, mid - 1)
        mid_root.right = self.traversal(nums, mid + 1, right)
        return mid_root

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

递归

思路：

1.确定递归参数和返回值：输入root，输出不需要递归函数的返回值做什么操作了，要遍历整棵树，需要定义一个全局变量，保存上一个节点值

2.终止条件：遇空就终止。

3.单层逻辑：要右中左来遍历二叉树，中节点的处理逻辑就是让cur的数值加上前一个节点的数值。

class Solution:
    def creatTree(self, root: Optional[TreeNode]):
        if not root:
            return None
        self.creatTree(root.right)
        root.val += self.pre
        self.pre = root.val
        self.creatTree(root.left)

    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.pre = 0
        self.creatTree(root)
        return root