二叉树的最近公共祖先

一、需求

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

• 百度百科中最近公共祖先的定义为：“对于有根树 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 为不同节点且均存在于给定的二叉树中。

二、后序遍历(DFS)

2.1  思路分析

1. 若root是p、q的公共祖先，则只有三种情况：①p、q在root的左右子树，分列root的异侧；②p = root，q在root左或右子树中；③q = root，p在root的左或右子树中；
2. 考虑利用后序遍历，自底向顶回溯，最后返回公共祖先root(最大深度)即可；
3. 首先是初始化，当root == null时，或root == p时，或root == q时，这些情况都要返回root；
4. 步骤3不成立，说明当前root不是公共祖先，递归root.left，root.right，令递归左子树的结果为left，递归右子树的结果为right；
5. ①left、right同时为null，说明没有找的公共祖先，直接返回null；②left == null，right != null，说明p，q在左子树，直接返回left；③left != null，right == null，说明p，q在右子树，直接返回right；④left != null，right != null，说明p，q分别在root的左、右子树，此时返回root；

2.2  代码实现

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root == null || root == p || root == q) return root;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left == null && right == null) return null;
        if(left != null && right != null) return root;
        if(left == null && right != null) return right;
        if(left != null && right == null) return left;
        return root;
    }
}

2.3  代码优化

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root == null || root == p || root == q) return root;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left == null) return right;
        if(right == null) return left;
        return root;
    }
}

2.4  复杂度分析

• 时间复杂度为O(N)，其中N为二叉树的节点数，最差情况下，需要递归遍历所有的节点；
• 空间复杂度为O(N)，最差情况下，二叉树退化为链表，递归的深度可达N；

三、学习地址

作者：Krahets

链接：https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof/solution/mian-shi-ti-68-ii-er-cha-shu-de-zui-jin-gong-gon-7/