题目链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
示例 1:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。
示例 2:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉搜索树中。
其实很简单,注意要求是返回结点,而不是值。因为是二叉搜索树,所以一个结点的左子树所有结点的值都小于等于该点的值,左子树所有结点的值都大于等于该点的值,原题目说明中第一条“所有节点的值唯一”那么上述不存在等于的情况。也就是说,对于根节点,如果任一结点的值小于其值,那么该结点位于其左子树中,大于则位于右子树中。如果两个结点分别位于根节点的左、右子树中(即根节点的值介于两个结点的值中间),显而易见他们的最近公共祖先是根节点;否则两个结点同位于根节点的左、右子树中(即两个结点都比根节点值大或小),那么两个结点的最近公共祖先也位于左、右子树中,并且最近公共祖先的值介于两个结点的值之间。
算法可通过递归/迭代来完成,输入为子树的根结点root,两个结点p和q,设vallow为p和q两个结点值的最小值,valhigh为最大值,则:
- 如果root的值大于等于vallow但小于等于valhigh的值,说明介于两者值之间,两个结点分别位于其左右子树,那么root即他们的最近公共祖先;
- 否则如果root的值大于valhigh的值,说明两个结点的值都比root的值小,即都位于左子树中,则进入root的左子树中寻找;
- 否则root的值小于vallow的值,说明两个结点的值都比root的值大,即都位于右子树中,则进入root的右子树中寻找。
情况1中两边取等号的原因为最近公共祖先节点可以为节点本身,所以如果存在相等的情况,说明两个结点中有一个结点为根节点。
递归方法:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
int vallow=min(p->val,q->val);
int valhigh=max(p->val,q->val);
if (vallow<=root->val && valhigh>=root->val)
return root;
else if(valhigh<root->val)
return lowestCommonAncestor(root->left,p,q);
else
return lowestCommonAncestor(root->right,p,q);
}
};
迭代方法:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
TreeNode* r=root;
int vallow=min(p->val,q->val);
int valhigh=max(p->val,q->val);
while(r){
if (vallow<=r->val && valhigh>=r->val)
break;//其实这行可以直接return r的,但是不通过
else if(valhigh<r->val)
r=r->left;
else
r=r->right;
}
return r;
}
};