一、题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
二、思路分析
注:思路分析中的一些内容和图片参考自力扣各位前辈的题解,感谢他们的无私奉献
思路
祖先的定义: 若节点 p p p 在节点 r o o t root root 的左(右)子树中,或 p = r o o t p = root p=root,则称 r o o t root root 是 p p p 的祖先
最近公共祖先的定义: 设节点 r o o t root root 为节点 p p p 和 q q q 的某公共祖先,若其左子节点 r o o t . l e f t root.left root.left 和右子节点 r o o t . r i g h t root.right root.right 都不是 p p p 和 q q q 的公共祖先,则称 r o o t root root 是最近的公共祖先。
根据以上定义,若 r o o t root root 是 p p p, q q q 的最近公共祖先,则只可能为以下情况之一:
p p p 和 q q q 在 r o o t root root 的子树中,且分列 r o o t root root 的异侧(即分别在左、右子树中)
p = r o o t p = root p=root,且 q q q 在 r o o t root root 的左或右子树中
q = r o o t q = root q=root,且 p p p 在 r o o t root root 的左或右子树中
考虑通过递归对二叉树进行先序遍历,当遇到节点 p p p 或 q q q 时返回。从底至顶回溯,当节点 p p p、 q q q 在节点 r o o t root root 的异侧时,节点 r o o t root root 即为最近公共祖先,则向上返回 r o o t root root。
递归解析:
①终止条件:
----当越过叶节点,则直接返回null
----当root
等于p
、q
,则直接返回root
②递推工作:
----开启递归左子节点,返回值记为left
----开启递归右子节点,返回值记为right
③返回值:根据left
和right
,可展开为四种情况:
----当left
和right
同时为空:说明root
的左/右子树中都不包含p
、q
,返回null
----当left
和right
同时不为空:说明p
、q
分列在root
的异侧(分别在左/右子树),因此root
为最近公共祖先,返回root
----当left
为空,right
不为空:说明p
、q
都不在root
的左子树中,直接返回right
。具体可分为两种情况:
--------p
、q
其中一个在root
的右子树中,此时right
指向p
(假设为p
)
--------p
、q
两节点都在root
的右子树中,此时的right
指向最近公共祖先节点
----当left
不为空,right
为空:与上面情况同理
案例分析:
复杂度分析:
时间复杂度 O ( N ) \rm{O(N)} O(N):其中N
为二叉树节点数,最差情况下,需要递归遍历树的所有节点
空间复杂度 O ( N ) \rm{O(N)} O(N):最差情况下,递归深度达到N
,系统使用O(N)
大小的额外空间
三、整体代码
整体代码如下
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeNode* lowestCommonAncestor(struct TreeNode* root, struct TreeNode* p, struct TreeNode* q){
if(!root || root->val == p->val || root->val == q->val) return root;
struct TreeNode* left = lowestCommonAncestor(root->left, p, q);
struct TreeNode* right = lowestCommonAncestor(root->right, p, q);
if(!left) return right;
if(!right) return left;
return root;
}
运行,测试通过