同剑指offer 68-II 二叉树的最近公共祖先/lcr 194
题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
题解
题解部分转载自代码随想录
此题与面试题68-I二叉搜索树的最近公共祖先的不同在于,二叉搜索树有特点left.val<root.val<right.val,所以可以很容易定位出p,q的位置。
本题需要判断左右子树的返回值是否为空判断,p,q是在左子树还是右子树中。p在左子树中,q在右子树中,根节点为父节点;或者是p为根节点,q为子节点,则p为根节点,如果左右子树均为空,则返回null
递归解析:
1. 确定递归函数的参数和返回值类型
因为要返回parent,所以返回值为TreeNode类型,返回的值为找到的p或q
注意:如果有时返回值需要为TreeNode或者为String,而我们需要遇到null值return的情况,这时我们可以将dfs返回值设为void以便于返回,用全局变量存储需要的return的TreeNode或者String
2. 终止条件(空节点需要特判):
当越过叶节点,则直接返回Null
当root等于p,q,则直接返回root
如果q在p的子树中,只找到p就提前返回了,会对结果有影响吗? 不会,因为如果q如果为p的子节点,那么p就是父节点
3.(本层节点操作)&&左右子树
开启递归左子节点,返回值记为left
开启递归右子节点,返回值记为right
4. 返回值: 根据left和right,可展开为三种情况:
1.当left和right同时为空,说明root的左右子树中都不包含p,q返回null。
2.当left和right同时不为空,说明p,q分列在root的左右两侧,因此root为最近公共祖先,返回root。
3.当left为空,right不为空:p,q都不在root的左子树中,则返回right,此时p,q两节点都在root的右子树中,此时的right指向最近公共祖先节点。
问题一:二叉树进行递归搜索时,如何区分要搜索一条边,还是搜索整个树呢?
搜索一条边的写法:(只需要找满足条件的某个值)
if (递归函数(root->left)) return ;
if (递归函数(root->right)) return ;
搜索整个树写法:(需要结合左右子树的结果判断返回值)
left = 递归函数(root->left); // 左
right = 递归函数(root->right); // 右
left与right的逻辑处理; // 中
- 在递归函数有返回值的情况下:如果要搜索一条边,递归函数返回值不为空的时候,立刻返回(如在二叉搜索树中搜索值为target的节点)
- 如果搜索整个树,直接用一个变量left、right接住返回值,这个left、right后序还有逻辑处理的需要,也就是后序遍历中处理中间节点的逻辑
注意:左或右子树的处理基本就是root=left||right, root=left&&right两种类型(如二叉树的最大深度就是1+max(左或右),本题就是root根据left&&right的结果判断)。
问题二:为什么left为空,right不为空,目标节点通过right返回呢?
图中节点10的左子树返回null,右子树返回目标值7,那么此时节点10的处理逻辑就是把右子树的返回值(最近公共祖先7)返回上去!
代码实现
//left和right不为空,返回root。否则若left不为空返回left的返回值,若right不为空返回right的返回值
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
//边界
if(root==null||p==null||q==null) return null;
//本层操作
if(root.val==p.val||root.val==q.val) 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&&right==null) return left;
else return right;
}
}