二叉树15:二叉树的最近公共祖先

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

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

要想找到两个节点的最近公共祖先节点,我们可以从两个节点往上找,每个节点都往上 走,一直走到根节点,那么根节点到这两个节点的连线肯定有相交的地方,如果是从上 往下走,那么最后一次相交的节点就是他们的最近公共祖先节点。我们就以找6和7的最 近公共节点来画个图看一下

我们看到6和7公共祖先有5和3,但最近的是5。我们只要往上找,找到他们第一个相同 的公共祖先节点即可,但怎么找到每个节点的父节点呢,我们只需要把每个节点都遍历 一遍,然后顺便记录他们的父节点存储在Map中。我们先找到其中的一条路径,比如 6→5→3,然后在另一个节点往上找,由于7不在那条路径上,我们找7的父节点是2,2 也不在那条路径上,我们接着往上找,2的父节点是5,5在那条路径上,所以5就是他们 的最近公共子节点。

其实这里我们可以优化一下,我们没必要遍历所有的结点,我们一层一层的遍历(也就 是BFS),只需要这两个节点都遍历到就可以了,比如上面2和8的公共结点,我们只需 要遍历到第3层,把2和8都遍历到就行了,没必要再遍历第4层了。

看递归的代码:

    public TreeNode lowestCommonAncestor(TreeNode cur, TreeNode p, TreeNode q) {
        if (cur == null || cur == p || cur == q)
            return cur;
        TreeNode left = lowestCommonAncestor(cur.left, p, q);
        TreeNode right = lowestCommonAncestor(cur.right, p, q);
        //如果left为空,说明这两个节点在cur结点的右子树上,我们只需要返回右子树查找的结果即可 if (left == null)
        return right; //同上
        if (right == null)
            return left;
//如果left和right都不为空,说明这两个节点一个在cur的左子树上一个在cur的右子树上,
// 我们只需要返回cur结点即可。
        return cur;
    }

对于这类稍微复杂一些的题目,代码不见得很复杂,我们会递归写法就足够了,关键是自己必须能想清楚为什么这么做可以。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

纵横千里,捭阖四方

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值