Leetcode【二叉树的最近公共祖先】

题目描述:

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先.
最近公共祖先的定义为：
对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x;
满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）

测试用例:

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

思路1（利用先序遍历）

- 先判断root是否为空,为空直接返回null；
- 如果root==p或者q,直接返回root;
- 递归遍历并判断左子树是否包含p和q,包含则递归调用找公共祖先的函数;
- 如果均不包含就递归遍历并判断右子树中是否包含p和q,包含则递归调用找	
  公共祖先的函数;

代码如下:

public class Solution {
    private boolean search(TreeNode root,TreeNode tmp){
        if (root == null){
            return false;
        }
        if (root == tmp){
            return true;
        }
        //递归判断左子树中是否含有tmp
        if (search(root.left,tmp)){
            return true;
        }
        return search(root.right,tmp);
    }
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //判断参数的合法性
        if (root == null){
            return null;
        }
        if (p == root || q == root){
            return root;
        }
        //判断p是否在左子树中
        boolean pInLeft = search(root.left,p);
        //判断q是否在左子树中
        boolean qInLeft = search(root.left,q);
        if (pInLeft && qInLeft){
            return lowestCommonAncestor(root.left,p,q);
        }
        if (!pInLeft && !qInLeft){
            return lowestCommonAncestor(root.right,p,q);
        }
        return root;
    }
}

思路2（利用后序遍历）

 -  先判断root是否为空,为空直接返回null；
 - 定义一个全局变量lca表示公共祖先;
 - 先判断左子树中是否含有p或者q,有则返回1,无则返回0，记作left；
 - 判断右子树中是否含有p或者q,有则返回1,无则返回0，记作right；
 - 判断根节点是否为p或者q,是则返回1,不是则返回0，记作mid;
 - 若left+right+mid == 2,则将此时的root赋值给lca；
 - 若left+right+mid <= 0,证明没有公共祖先;

代码如下:

public class Solution1 {
    private TreeNode lca = null;
    private boolean findNode(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null){
            return false;
        }
        int left = findNode(root.left,p,q)?1:0;
        int right = findNode(root.right,p,q)?1:0;
        int mid = (root == p || root == q)?1:0;
        if (left+right+mid == 2){
            lca = root;
        }
        return (left+right+mid)>0;
    }
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null){
            return null;
        }
        findNode(root,p,q);
        return lca;
    }


}