给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
思路:最近公共祖先的条件有两种
一:该节点左右子树分别都含有p,q
二:给节点等于q或p,并且它至少有一颗子树有p或q
代码如下:
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root==null) return null;
boolean left=dfs(root.left,p,q);
boolean right=dfs(root.right,p,q);
if(left&&right||(root.val==p.val||root.val==q.val)&&(left||right)) return root;
TreeNode node=lowestCommonAncestor(root.left,p,q);
if(node!=null) return node;
else return lowestCommonAncestor(root.right,p,q);
}
public boolean dfs(TreeNode root,TreeNode p,TreeNode q){//这个函数表示该节点有q或p
if(root==null) return false;
if(root.val==p.val||root.val==q.val) return true;
return dfs(root.left,p,q)||dfs(root.right,p,q);
}
}