剑指offer68-II:二叉树的最近公共祖先(后序遍历--回溯)

题目

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

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

例如,给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]

示例 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。因为根据定义最近公共祖先节点可以为节点本身。
说明:所有节点的值都是唯一的。p、q 为不同节点且均存在于给定的二叉树中。

解题思路

二叉树寻找公共祖先的方法是自底向上查找,那么应该采用后序遍历来完成自然的回溯过程,来最先处理叶子节点。

如果找到一个节点,发现左子树出现节点p,右子树出现节点q,或者左子树出现节点q,右子树出现节点p,那么该节点就是节点p和q的最近公共祖先。使用后序遍历,回溯的过程,就是从底向上遍历节点,一旦发现符合这个条件的节点,就是最近公共节点了。

递归三部曲:

  • 确定递归函数返回值以及参数:通过是否找到节点p或者q,返回bool类型即可,但是还要返回公共节点,那么如果遇到p或者q,就把q或者p返回,返回值不为空,就说明找到了q或者p。
  • 确定终止条件:如果找到了节点p或者q,或者遇到空节点,就返回。
  • 确定单层递归逻辑:本题函数有返回值,是因为回溯的过程需要递归函数的返回值做判断,但是本题依然要遍历树的所有节点。

如果递归函数有返回值,如何区分要搜索一条边,还是搜索整个树呢?

搜索一条边的写法:

if (递归函数(root->left)) return ;

if (递归函数(root->right)) return ;

搜索整个树写法:

left = 递归函数(root->left);
right = 递归函数(root->right);
left与right的逻辑处理;

可以看出区别:在递归函数有返回值的情况下,如果要搜索一条边,递归函数返回值不为空的时候,立刻返回,如果搜索整个树,直接用一个变量left、right接住返回值,这个left、right后续还有后序遍历中处理中间节点的逻辑处理的需要(回溯),并且不能立刻返回,而是要等left和right逻辑处理完之后才能返回。

完整流程图:

 代码:

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if not root or root == p or root == q: return root  //找到了节点p或者q,或者遇到空节点
        left = self.lowestCommonAncestor(root.left,p,q)  //左
        right = self.lowestCommonAncestor(root.right,p,q)  //右
        if left and right: return root  //中: left和right不为空,root就是最近公共节点
        elif left and not right: return left  //目标节点是通过left返回的
        elif not left and right: return right  //目标节点是通过right返回的
        else: return None  //没找到

其中,如果left和right都不为空,说明此时root就是最近公共节点;如果left为空,right不为空,就返回right,说明目标节点是通过right返回的,反之亦然。

那么为什么left为空,right不为空,目标节点通过right返回呢?

图中节点10的左子树返回null,右子树返回目标值7,那么此时节点10的处理逻辑就是把走字数的返回值(最近公共祖先7)返回上去。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值