算法通关村-----寻找祖先问题

最近公共祖先

问题描述

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

问题分析

按照前序遍历的顺序遍历二叉树，对于遍历到的当前节点root，判断其与要寻找公共祖先的两个节点p和q的关系,如果root=p或者root=q则，root是p和q的最近公共祖先，否则，在root的左右子树分别寻找p和q，如果p和q在root的左子树和右子树，则root是p和q的最近公共祖先，如果p和q均在root的左子树或者右子树，则在root的左子树或者右子树继续递归寻找最近公共祖先

代码实现

private TreeNode res = null;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    dfs(root,p,q);
    return res;
}

public boolean dfs(TreeNode root, TreeNode p,TreeNode q){
    if(root==null){
        return false;
    }
    boolean left = dfs(root.left,p,q);
    boolean right = dfs(root.right,p,q);
    if(left&&right){
        res = root;
    }
    if(root == p || root ==q){
        res = root;
    }
    return left || right || root==p || root==q;
}