题目描述:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
方法一:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
//如果为空就直接返回
if (root == nullptr) return nullptr;
//如果当前节点为p或者q,就直接返回这个节点
if (root->val == p->val || root->val == q->val) return root;
TreeNode* curleft = lowestCommonAncestor(root->left, p, q);
TreeNode* curright = lowestCommonAncestor(root->right, p, q);
//如果curleft或curright非空,说明找到了p或者q或者题目所要求的最近公共祖先节点
//也有可能是curleft和curright都非空,这说明什么节点都没找到,返回nullptr
if (curleft == nullptr) return curright;
if (curright == nullptr) return curleft;
//如果curleft和curright都非空,说明root就是p和q的最近公共祖先节点
if (curleft && curright) return root;
//至少返回一个nullptr
return nullptr;
}
};
前几天的【二叉树的最近公共祖先】一题的方法,重写一遍加深印象,第二次写完后感觉自己的理解比前几天时深刻了不少。
注意判断两个节点是否相等时,上一题可以直接使用root==p,但这题由于代码的缘故只能使用root.val==p.val。
方法二:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == nullptr) return nullptr;
if (root == p || root == q) return root;
//若根节点的值小于p和q的值,说明最近公共祖先在右子树中
if (root->val < p->val && root->val < q->val)
{
return lowestCommonAncestor(root->right, p, q);
}
//若根节点的值大于p和q的值,说明最近公共祖先在左子树中
else if (root->val > p->val && root->val > q->val)
{
return lowestCommonAncestor(root->left, p, q);
}
//若以上情况都不符合,说明根节点的值介于p和q的值之间,必定是最近公共祖先
else
{
return root;
}
return nullptr;
}
};
这个方法利用平衡二叉树的原理实现,p和q的最近公共祖先要么就刚好为p或q,要么其值就介于p和q的值之间。
所以对于这棵树中的每个节点只有几种情况:
①该节点恰好为p或q,跟上一题一样,若递归的过程中首先遇到了p或q,那么说明其必定是最近公共祖先
②根节点的值小于p和q的值,根据平衡二叉树的定义,说明最近公共祖先在右子树中
③根节点的值大于p和q的值,根据平衡二叉树的定义,说明最近公共祖先在左子树中
④根节点的值恰好在p和q的值之间,这说明p和q肯定分别在该节点的两个子树中,所以该节点就是最近公共祖先
方法二:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
while ((root->val - p->val) * (root->val - q->val) > 0)
{
root = root->val < p->val ? root->right : root->left;
}
return root;
}
};
太秀了,完全想不到还有这么简洁好看的方法!
关于(root->val - p->val) * (root->val - q->val) > 0这个条件,上述所列的情况②和情况③都满足,直到找到情况①(==0的情况)或者情况④(<0的情况),就说明此时的root是最近的公共祖先了。