给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例 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
因为根据定义最近公共祖先节点可以为节点本身。
示例 3:
输入:root = [1,2], p = 1, q = 2 输出:1
提示:
[2, 105]
内。-109 <= Node.val <= 109
Node.val
互不相同
。p != q
p
和 q
均存在于给定的二叉树中。这道题的递归算法并不好理解,需要结合讲解和代码结合着多理解几遍。
1、如果root等于p或q,说明root本身就是最近公共祖先,直接返回
2、如果root为NULL则说明到了叶子节点,直接返回
3、分别查询左节点和右节点是否能够查到p或q
若p左q右,或p右q左,说明root就是最近公共祖先
若都在一边,那么只要继续查这一边就可以了
struct TreeNode* lowestCommonAncestor(struct TreeNode* root, struct TreeNode* p, struct TreeNode* q) {
if (root == NULL || root == p || root == q) return root;
struct TreeNode *left = lowestCommonAncestor(root->left, p, q);
struct TreeNode *right = lowestCommonAncestor(root->right, p, q);
if (left == NULL) return right;
if (right == NULL) return left;
return root;
}
力扣题目网址:力扣——全球极客挚爱的技术成长平台
萌新不定期在互联网上随地乱丢一些赛博垃圾,还望拨冗批评斧正~