题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
如图:
示例 1: 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出: 3
解释: 节点 5 和节点 1 的最近公共祖先是节点 3。
示例 2:
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出: 5
解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。
说明:
所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。
通过次数66,846
提交次数107,744
思路
由题我们可以知道,p,q均在给定的二叉树中。要想找到最近的共同祖先,可以用递归来解决。
首先,我们需要找到递归的出口:
有以下几种情况:
1.没找到p,q。对当前root进行左右子树递归查找p,q。如果递归到root==null则返回root,表示当前递归结束。
2.找到p,q。直接返回当前root。
接着我们需要根据递归结果进行判断:
1.如果左右子树递归的值都不为空,表示左右子树均找到了p,q,此时root就为他们的最近共同祖先。
2.如果左子树不为空,右子树为空,表示左子树找到了第一个p,q,此时下面必然存在另外一个待找的节点。返回当前已找到的节点即为p,q的共同祖先。(因为根据定义最近公共祖先节点可以为节点本身)
3.如果右子树不为空,左子树为空,表示右子树找到了第一个p,q,此时下面必然存在另外一个待找的节点。返回当前已找到的节点即为p,q的共同祖先。(因为根据定义最近公共祖先节点可以为节点本身)
4.如果左右子树均为空,p,q不在数中,返回空。
下面附上java代码:
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
//递归结束的条件
if (root==null||root==p||root==q){
return root;
}
//递归左子树
TreeNode left = lowestCommonAncestor(root.left,p,q);
//递归右子树
TreeNode right = lowestCommonAncestor(root.right,p,q);
//判断结果的条件
if (left!=null&&right!=null){
return root;
}else if (left!=null){
return left;
}else if (right!=null){
return right;
}
return null;
}
}
小结:
在树的结构中,我们经常需要采用递归的思想对树进行遍历。此时我们需要关注的就是递归的出口。只要找到出口,递归函数就十分容易上手!此外,在树的遍历中有两个十分常见的算法:DFS(深度优先遍历)和BFS(广度优先遍历)。希望看到此篇文章的小伙伴们去了解一下这两个遍历树的算法!