给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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
均存在于给定的二叉树中。
以上是普通二叉树的?如果是二叉搜索树又怎么处理呢?
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL ) return NULL; //这两个打破递归循环的逻辑,也就是函数的出口
if(root == q || root== p) return root; //说明找到了一个p or q,返回当前的一个根节点,而不是null
TreeNode* left = lowestCommonAncestor(root->left,p,q);
TreeNode* right =lowestCommonAncestor(root->right,p,q);
//中
if(left!=NULL && right!= NULL) return root;// 进入递归的是root->left,返回的也是root->left,因此当满足条件是返回root
//下面这个逻辑是递归逻辑
if(left!=NULL && right ==NULL) return left;
else if(right !=NULL && left == NULL) return right;
else return NULL;
}
};
我给它进行了注释!主要是围绕递归三部曲!
1.返回值和参数
2.打破递归的出口(条件)
3.单层的循环条件逻辑
而对于搜索二叉树的做法!是这样的!这样是错的,我陷入了一个错误的想法!
class Solution {
public:
//先写一种情况,q->val < q->val 既 公共祖先的值在 [p,q]之间
TreeNode* first_action(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL) return NULL ;
if(root == q || root == p) return root;
TreeNode* left = NULL;
TreeNode* right = NULL;
if(root->left != NULL && root->left->val >= p->val && root->left->val <= q->val) {
left = first_action(root->left, p, q);
} //进入左子树递归的条件
if(root->right != NULL && root->right->val >= p->val && root->right->val <= q->val) {
right = first_action(root->right, p, q);
} //进入右子树递归的条件
if(left != NULL && right != NULL) return root; //中
// 单层
if(left != NULL && right == NULL) return left;
else if(left == NULL && right != NULL) return right;
else return NULL;
}
TreeNode* sencen_action(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL) return NULL ;
if(root == q || root == p) return root;
TreeNode* left = NULL;
TreeNode* right = NULL;
if(root->left != NULL && root->left->val >= q->val && root->left->val <= p->val) {
left = sencen_action(root->left, p, q);
} //进入左子树递归的条件
if(root->right != NULL && root->right->val >= q->val && root->right->val <= p->val) {
right = sencen_action(root->right, p, q);
} //进入右子树递归的条件
if(left != NULL && right != NULL) return root; //中
// 单层
if(left != NULL && right == NULL) return left;
else if(left == NULL && right != NULL) return right;
else return NULL;
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
TreeNode* node ;
if(p->val < q->val) node=first_action(root,p,q);
else node=sencen_action(root,p,q);
return node;
}
};
只要从上到下去遍历,遇到 cur节点是数值在[p, q]区间中则一定可以说明该节点cur就是p 和 q的公共祖先。 那问题来了,一定是最近公共祖先吗?所以当我们从上向下去递归遍历,第一次遇到 cur节点是数值在[q, p]区间中,那么cur就是 q和p的最近公共祖先。因为对应二叉搜索树来说,从上到下去遍历,只要第一次出现在这个区间的节点就一定是最近的公共祖先!因为他是有序的!
左边比根节点小,右边比根节点大!所以说只要出现了在[q,p区间里],要是往左边走了,就会出现当前的节点大小值为范围[p,q),导致再也无法再遇见q了,同理往右边走也是同一个道理!!,往右边走的话,当前的节点大小值为范围(q,q+1),导致公共祖先的节点【p,q】无法在相遇!因此,只要明白了这一点,我相信手到擒来!
而迭代法会更简单!
迭代代码如下:一直找符合这个范围的root!!!
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
while(root) {
if (root->val > p->val && root->val > q->val) {
root = root->left;
} else if (root->val < p->val && root->val < q->val) {
root = root->right;
} else return root;
}
return NULL;
}
};