day23代码随香录

669. 修剪二叉搜索树

 

思路:

      这是一个特例,当发现0不符合我们的范围区间的时候,不能直接return none,还要去考虑0的右子树,这里可以直接将0的右子树赋值给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:
            right = self.trimBST(root.right,low,high)
            return right
        if root.val > high:
            left = self.trimBST(root.left,low,high)
            return left 
        root.left = self.trimBST(root.left,low,high)
        root.right = self.trimBST(root.right,low,high)
        return root

        

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

 

 思路:本质就是寻找分割点,分割点作为当前节点,然后递归左区间和右区间。分割点就是数组中间位置的节点。那么为问题来了,如果数组长度为偶数,中间节点有两个,取哪一个?

取哪一个都可以,只不过构成了不同的平衡二叉搜索树

这里定义的是左闭右闭区间,在不断分割的过程中,也会坚持左闭右闭的区间,这又涉及到我们讲过的循环不变量。 

首先取数组中间元素的位置,不难写出 mid = (left + right) / 2;这么写其实有一个问题,就是数值越界,例如left和right都是最大int,这么操作就越界了

所以可以这么写:int mid = left + ((right - left) / 2);

但本题leetcode的测试数据并不会越界,所以怎么写都可以。但需要有这个意识!

取了中间位置,就开始以中间位置的元素构造节点,代码:TreeNode* root = new TreeNode(nums[mid]);

接着划分区间,root的左孩子接住下一层左区间的构造节点,右孩子接住下一层右区间构造的节点。

代码:

        

def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        def helper(left, right):
            if left > right:
                return None
            # 总是选择中间位置左边的数字作为根节点
            mid = (left + right) // 2
            root = TreeNode(nums[mid])
            root.left = helper(left, mid-1)
            root.right = helper(mid + 1, right)
            return root

        return helper(0, len(nums) - 1)
538. 把二叉搜索树转换为累加树

 

题目都给看蒙了 实际上就是 不断累加比自己数值大的节点,也就是倒序的中序遍历 右中左的顺序遍历

 

pre指针的使用技巧,我们在二叉树:搜索树的最小绝对差 (opens new window)二叉树:我的众数是多少? (opens new window)都提到了,这是常用的操作手段。

  • 递归函数参数以及返回值

这里很明确了,不需要递归函数的返回值做什么操作了,要遍历整棵树。

同时需要定义一个全局变量pre,用来保存cur节点的前一个节点的数值,

  代码:

        

def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.pre = 0  # 记录前一个节点的数值
        self.traversal(root)
        return root
    def traversal(self, cur):x
        if cur is None:
            return        
        self.traversal(cur.right)
        cur.val += self.pre
        self.pre = cur.val
        self.traversal(cur.left)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值