236. 二叉树的最近公共祖先

236. 二叉树的最近公共祖先

难度中等1085收藏分享切换为英文接收动态反馈

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

 

示例 1：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3
解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出：5
解释：节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

示例 3：

输入：root = [1,2], p = 1, q = 2
输出：1

 

提示：

• 树中节点数目在范围 [2, 105] 内。
• -109 <= Node.val <= 109
• 所有 Node.val 互不相同 。
• p != q
• 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':

        # ans = []

        # def dfs(path,node,target):  

        #     path.append(node)

        #     if node.val == target:

        #         ans.append(path+[node])

        #         return 

        #     else:

        #         if node.left:

        #             dfs(path,node.left,target)

        #             path.pop()

        #         # print(path,target)

        #         if node.right:

        #             dfs(path,node.right,target)

        #             path.pop()

        # dfs([],root,p.val)

        # dfs([],root,q.val)

        # n = len(ans[0])

        # print(ans)

        # for i in range(n):

        #     if ans[0][i] != ans[1][i]:

        #         return ans[0][i-1]

        # print(n,ans[0][n-1])

        # return ans[0][n-1]

        

        if not root or root == p or root == q: return root

        left = self.lowestCommonAncestor(root.left, p, q)

        right = self.lowestCommonAncestor(root.right, p, q)

        if not left: return right

        if not right: return left

        return root

 

            https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/solution/236-er-cha-shu-de-zui-jin-gong-gong-zu-xian-hou-xu/