给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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。
思路:利用二叉搜索树中序遍历有序的特性,我们只要在[p,q]区间中查找两者的最近公共祖先既可。
- 确定递归函数的参数列表和返回值:需要的参数有根节点,p节点和q节点,返回值为两者的最近公共祖先。
- 确定终止条件:遇到空节点返回或者可以不写,因为题目中说p和q均存在,所以一定会遇到p和q。
- 确定单层递归逻辑:如果p和q节点的值均小于当前节点值,说明p和q均在当前节点左侧,返回左子树的查找结果;如果p和q节点的值均大于当前节点值,说明p和q在当前节点右侧,返回右子树的查找结果;否则,说明当前节点值在p和q之间,说明当前节点就是最近公共祖先,返回当前节点。
代码:
class Solution { //235. 二叉搜索树的最近公共祖先
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root->val > p->val && root->val > q->val)
return lowestCommonAncestor(root->left, p, q);
else if (root->val < p->val && root->val < q->val)
return lowestCommonAncestor(root->right, p, q);
else return root;
}
};
参考资料:代码随想录