二叉树的最近公共祖先

还可以再优化,使用递归两步解决,第一步找出p的所有父节点,第二步从p的最后一个父节点开始依次找q

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':

        def commonAncestor(root, q, res):
            if root is None:
                return False
            if root.val == q.val:
                res.append(root)
                return True
            if root.left is not None:
                if commonAncestor(root.left, p, res):
                    res.append(root)
                    return True
            if root.right is not None:
                if commonAncestor(root.right, p, res):
                    res.append(root)
                    return True

        def commonAncestorSplit(root, q):
            if root is None:
                return False
            if root.val == q.val:
                return True
            if root.left is not None:
                if commonAncestorSplit(root.left, q):
                    return True
            if root.right is not None:
                if commonAncestorSplit(root.right, q):
                    return True
            return False

        res = []
        commonAncestor(root, p, res)
        for r in res:
            if commonAncestorSplit(r, q):
                return r

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值