235. 二叉搜索树的最近公共祖先以及236.二叉树的最近公共祖先

本质上两道题的解法可以一样,但是二叉搜索树右个性质是根节点的值大于左节点的值小于右节点的值,将该特性用上可以提升效率
1、二叉搜索树的最近公共祖先
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。
输入root为层序遍历的结果,可直接还原完全二叉树
解法:利用上述性质直接求解

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        #直接利用性质,二叉搜索树的中序遍历是升序的。判断root.val和p.val,q.val的关系
        if ((root.val-p.val)*(root.val-q.val)<=0):
            return root
        if root.val>p.val:
            return self.lowestCommonAncestor(root.left, p, q)
        else:
            return self.lowestCommonAncestor(root.right, p, q)

注意root的判断过程中一定要加等号,此时表明p或者q就是root。

2、二叉树的最近公共祖先
本质重点一样:进行遍历,判断p,q和当前root的位置。
类似于二叉树的后续遍历,先判断左右节点的位置,再和root进行关系对比
1)迭代终止条件:root=none, root=p or root=q, return root
2)进行迭代到下一个节点
3)节点返回进行判断,如果分别左右都有则返回root
如果只在左边,返回左节点的返回值
如果只在右边,返回右节点的返回值
如果都没有,返回空

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        #如果root为空,或者p或q就是root,则返回root
        if root ==None or root ==p or root ==q:
            return root
        #往下一个节点搜
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        #如果分散在左右:则返回root,如果只有左边有p或者q则返回左边,右边也是一样。如果都没有,则返回None
        if left and right:
            return root
        elif left:
            return left
        elif right:
            return right
        else:
            return None
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值