最低公共祖先

最低公共祖先

题目描述

给定一棵二叉树以及这棵树上的两个节点 o1 和 o2，请找到 o1 和 o2 的最近公共祖先节点。

代码

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) {
        // write code here
         if(root == null) return -1;//到底返回
         if(root.val == o1 || root.val == o2) return root.val;//得到最低公共祖先
         int left = lowestCommonAncestor(root.left, o1, o2);
         int right = lowestCommonAncestor(root.right, o1, o2);
         
         if(left==-1 && right==-1) return -1; //左右不存在o1、o2，且它们的最低公共祖先
         else if(left!=-1 && right==-1) return left; //右不存在o2，左存在o1或o2或它们的最低公共祖先
         else if(left==-1 && right!=-1) return right; //左不存在o2，右存在o1或o2或它们的最低公共祖先
         else return root.val; //最低公共祖先
    }
    
    
}