leetcode 236. 二叉树的最近公共祖先
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先(一个节点也可以是它自己的祖先)。
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
![6db9d551c58e31e738e22e2d3be053c7.png](https://i-blog.csdnimg.cn/blog_migrate/4fcdbf8950e0decd1182cfe9e878353f.jpeg)
示例 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。因为根据定义最近公共祖先节点可以为节点本身。
来源:力扣(LeetCode)
没有思路,借鉴文章
因为题目中给的数据一定存在公共祖先,所以要么是根节点,要么在左子树,要么在右子树。 - 如果根节点已经包含待求值中的某一个值,则根节点一定是最早公共祖先;
- 如果左子树中没有待求值,但右子树中返回了某一节点值,则可判断右子树中存在公共子节点,而递归操作就可以达到查找到最早公共子节点的作用
- 如果左子树中返回每一节点,而右子树中为空,与上述情况相同
- 如果左右子树均有返回节点,则可判断左右子树中应分别有一个待求节点,故此时的根节点一定是最早公共祖先。
# 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':
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 left == None:
return right
if right == None:
return left
return root
对于树的问题,记得用递归来思考问题