二叉树进阶题
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。(最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”) p 和 q 均存在于给定的二叉树中。
思路
自上而下的深度遍历DFS。理解下面这个图,然后看代码注释。
代码
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == null) return null;
if(root == p || root == q) {
return root;
}
TreeNode leftTree = lowestCommonAncestor(root.left,p,q);
TreeNode rightTree = lowestCommonAncestor(root.right,p,q);
if(leftTree != null && rightTree != null) {
return root;
}
if(leftTree == null && rightTree != null) {
return rightTree;
}
if(leftTree != null && rightTree == null) {
return leftTree;
}
if(leftTree == null && rightTree == null) {
return null;
}
return null;
}
|