LeetCode 二叉树的最近公共祖先

LeetCode 1、二叉搜索树的最近公共祖先

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

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

示例 1:

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

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。

说明:

所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-lcof

通过题意，我们可以直到一个节点的祖先的定义，那么需要求解两个节点p,q的最近的公共祖先有几种情况需要讨论:
①p、q两个节点位于root的两侧，那么这时候root就是他们的最近的公共祖先。怎样知道p、q两节点位于root的两侧呢？由于是一棵二叉搜素树，那么就有左子树的值都小于当前节点的值，并且右子树的值大于等于当前节点的值。所以在明确知道p的值小于q的值的前提下，只要root的值大于p的值，并且小于q的值，那么就说明两个节点位于不同侧。
②如果p = root，那么这时候q就在p的左子树或者右子树，此时最近公共祖先就是p
③如果q = root,那么这时候p就在q的左子树或者右子树，此时最近公共祖先就是q

对应的代码:

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //交换两个节点，从而保证p.val < q.val
        if(p.val > q.val){
            TreeNode tmp = p;
            p = q;
            q = tmp;
        }
        while(root != null){
            if(root.val < p.val) //如果根节点的值比p、q两个节点的值都要小，那么就说明p、q两个节点都在root的右子树
               root = root.right;
            else if(root.val > q.val) //如果根节点的值都比p、q两个节点的值都要大，那么说明p、q两个节点都在root的左子树位置
               root = root.left;
            else break;//如果p、q两个节点位于root的两个方向(即两个节点分处于root不同子树），或者其中一个节点是根节点，那么root就是最近的公共祖先
        }
        return root;
    }
}

LeetCode 二叉树的最近公共祖先

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

百度百科中最近公共祖先的定义为：“对于有根树 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。因为根据定义最近公共祖先节点可以为节点本身。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof

思路和上面题目基本类似，但是又有所不同：
①由于是一颗二叉树，所以并不存在当前节点的值大于它的左子树的值，当前节点的值小于等于右子树的所有节点的值的特性。因此，如果需要判断p、q两个顶点是否在root的两侧，怎样判断呢？我们分别在root的左右子树中进行查找p、q两个节点，因为p、q必定存在于二叉树中，并且节点是唯一的，所以两个子树都能找到，说明root就是他们的最近公共祖先。
②如果其中一个顶点是root的情况怎么判断呢？很明显，如果p、q中有一个作为root，假设p是root，那么另一个节点q必然是这个root节点p的子节点，此时是没有办法在root的兄弟节点中找到另一个节点的。

对应的代码:

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);//在左子树中查找p、q
        TreeNode right = lowestCommonAncestor(root.right,p,q);//在右子树中查找p、q
        /*
        注意不可以说left不为空的时候，就返回left，因为left不为空，只能说明
        可以说明在root的左子树中，可以找到一个节点，但是不能保证p、q两个节
        点就位于同一侧，会存在着另一个节点在root的右子树的情况，
        所以这样会发生报错
        if(left != null) 
           return left;
        */
        if(left == null){
        /*
        如果left为空，说明没有办法在左子树中找到p、q两个节点，
        那么就说明两个节点可能都在右子树，也可能都不在右子树，也有可
        能只有一个节点在右子树中,但是题目已经明确说明了p、q都存在于二叉树
        中，所以此时说明了两个节点都位于右子树，所以返回right
       */
            return right;
        }
        if(right == null){
            //如果left不为空，那么并且right为空，说明两个节点都在左子树
            return left;
        }
        return root;//如果两个节点left、right都不为空，说明在两侧，直接返回root
    }
}

运行结果: