面试题 04.08. 首个共同祖先(中等题)

1.题目描述:

设计并实现一个算法,找出二叉树中某两个节点的第一个共同祖先。不得将其他的节点存储在另外的数据结构中。
注意:这不一定是二叉搜索树。
链接:https://leetcode-cn.com/problems/first-common-ancestor-lcci/

2.题解思路:

从根节点出发,考虑p、q的情况:
(1)如果p 、q中有一个是根节点,那么它本身就是首个共同祖先;
(2)如果p、q分别在根节点的左右子树中,那么它本身也是首个共同祖先;
(3)如果p、q都同在根节点的左子树或右子树中,那就用递归的方法,将左子节点或右子节点作为根节点求解。

3.代码实现:

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //当root为空,就返回空
        if (root == null) {
            return null;
        }
        //当root是p或者q中的一个,满足(1)的条件,直接返回root
        if (root == p || root == q) {
            return root;
        }
        //在左子树中查找p和q
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        //在右子树中查找p和q
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        //如果left和right都不为空,说明p、q分别在左右子树上,满足(2)的条件,直接返回root
        if (left != null && right != null) {
            return root;
        }
        //如果左子树为空,说明p、q都在右子树中,将右子节点作为根节点,满足(3)
        if (left == null) {
            return right;
        }
        //如果右子树为空,说明p、q都在左子树中,将左子节点作为根节点,满足(3)
        if (right == null) {
            return left;
        }
        //如果左右子树都为空,说明p、q都不存在,本题不会出现这种情况
        return null;
    }

关于left和right,在递归过程中,要么值为null,要么就是我们找到的p或者q值。
当我们令 left = self.lowestCommonAncestor(root.left, p, q) 时,如果在左子树中找到了 p 或 q,left 会等于 p 或 q,同理,right 也是一样。

事实上,就是根据lowestCommonAncestor的返回值,来决定root往左子树还是右子树下移,直到最后某一个节点的left和right的值都不为空,那么它就是要求的首个共同祖先。

上边的代码可以进行简化:

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //当root为空,就返回空,当root是p或者q中的一个,满足(1)的条件,直接返回root
        if (root == null || root == p || root == q) {
            return root;
        }
        //在左右子树中查找p和q
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        return left != null ? right != null ? root : left : right;
    }
复杂度分析:
  • 时间复杂度:O(logn),最优的情况是O(1),最差的情况是O(logn)。
  • 空间复杂度:O(1),不需要额外开辟新的空间。


    竟然击败了双百的人,纪念一下
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 2.1.2 可并堆的定义 可并堆(Mergeable Heap)也是一种抽象数据类型,它除了支持优先队列的三个基本...
    Gitfan阅读 5,269评论 0 0
  • 目录 0.树0.1 一般树的定义0.2 二叉树的定义 1.查找树ADT 2.查找树的实现2.1 二叉查找树2.2 ...
    王侦阅读 12,119评论 0 3
  • 面试题3——数组中重复的数字 使用LinkedHashMap,有序存放。 面试题4——二维数组中的查找 首先选...
    做自己的Yang光阅读 3,493评论 0 0
  • 1. 树的概念 一个树由节点组成,这些节点包含根节点,父节点,子节点,兄弟节点;没有任何一个节点的树称为空树;如果...
    HChase阅读 11,606评论 0 34
  • 最近比较烦,用专业的方法来稳定心神,收效甚微。于是就毫不犹豫地使用了终极大招。 终极大招一:无糖可乐。吃甜食能够迅...
    行走的LILY阅读 1,117评论 0 0

友情链接更多精彩内容