题目
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
例如，给定如下二叉搜索树:
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 为不同节点且均存在于给定的二叉搜索树中。注意：本题与主站 235 题相同：LeetCode - Lowest Common Ancestor of a Binary Search Tree
解法：
- 本题给的二叉树为二叉搜索树，当节点值比 root 小，则在 root->left 上，反之则在 root->right 上
- 如果都在 root->left 上，则在左子树进行搜索，反之在右子树进行搜索
- 如果在 root 两侧，则 root 即为最近公共祖先
cpp
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(p->val > q->val) swap(p, q); // 保证 p 的值小于等于 q 的值
if(p->val <= root->val && q->val >= root->val) return root; // 如果 p、q 分别在 root 的两侧，说明 root 就是最近公共祖先
if(q->val < root->val) return lowestCommonAncestor(root->left, p, q); // 如果 p、q 都在 root 的左侧，则递归左子树
if(p->val > root->val) return lowestCommonAncestor(root->right, p, q); // 如果 p、q 都在 root 的右侧，则递归右子树
return nullptr; // 默认情况下返回空指针
}
};