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

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

示例 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 为不同节点且均存在于给定的二叉树中。

若root是p,q的最近公共祖先,则只可能为以下情况之一：

1. p和q在root的子树中，且分列root的异侧（即分别在左、右子树中）；
2. p=root,且q在root的左或右子树中；
3. q=root，且p在root的左或右子树中；

 

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
        
        //根节点为null 直接返回
        if(root == null){
            return 0;
        }
        
        //p=root,且q在root的左或右子树中；
        //q=root，且p在root的左或右子树中；
        //p或q其中一个是另一个的祖先结点，返回是祖先结点的那个
        if(root.val == o1 || root.val == o2){
            return root.val;
        }
        
        //p和q在root的子树中，且分列root的异侧（即分别在左、右子树中）
        // 在左子树寻找p和q的最近公共祖先
        int left = lowestCommonAncestor(root.left,o1,o2);
        
         // 在右子树寻找q和q的最近公共祖先
        int right = lowestCommonAncestor(root.right,o1,o2);
        
        //p和q分别位于他们最近公共结点的两侧
        // 如果p和q分属两侧，则根节点为最近公共祖先
        if(left!=0 && right!=0){
            return root.val;
        }
        
        // 如果左子树有值，则最近公共祖先在左子树，否则，在右子树
        //p或q中有一个为空，返回非空的那个
        if(left == 0){
            return right;
        }
         
        // 如果左子树有值，则最近公共祖先在左子树，否则，在右子树
        //p或q中有一个为空，返回非空的那个
        if(right == 0){
            return left;
        }
        
        return 0;
    }
}