题目来源: 236.二叉树的最近公共祖先
题目: 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
思路:递归
- 如果根结点为空,或者根结点等于p或q,则根结点即为其最近公共祖先。
- 如果根结点不为空,递归调用根结点的左右子树
- 如果左子树为空,说明p和q不在左子树中,则返回右子树。
- 如果右子树为空,说明p和q不在右子树中,则返回左子树。
- 如果左右子树都不为空,说明最近公共结点为根结点,返回根结点
解释:其实就是利用递归的思想,左右寻找p、q,把首先遇到p、q结点的结点返回,接着再判断p、q是否在同一个子树上,如果是则返回相应子树,不是的话公共祖先一定是根结点,一直递归判断。
python代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: 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 not left:
return right
if not right:
return left
return root
参考来源:【leetcode】剑指 Offer 68 - II. 二叉树的最近公共祖先(python3 版)
二叉树最近公共祖先(Python版)