剑指offer 68 - I. 二叉搜索树的最近公共祖先
题目描述
解题思路
这道题目的思路主要就是判断 p 和 q 的左右位置:
- 若 p,q 中有一个指向 root,则最近公共祖先就是root
- 若 p,q 分别在 root 树的两侧,则最近公共祖先就是root
- 若 p,q 均在 root 树的同一边子树中,则最近公共祖先就是 该子树的最近公共祖先
1. 普通递归
class Solution {
//定义:在以root为根节点的树中,找到p,q的最近公共祖先
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) return null;
if (root == p || root == q) return root;
//根据两种选择的不同情况判断
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
// 情况1:如果left和right都是空,则返回null
if (left == null && right == null) {
return null;
}
// 情况2:如果left和right各有p,q中的一个,则最近公共祖先就是root
if (left != null && right != null) {
return root;
}
// 情况3:如果p,q都在同一棵子树中,则返回该子树的最近公共祖先
return left == null ? right : left;
}
}
利用二叉搜索树的性质(推荐)
class Solution {
//定义:在以root为根节点的树中,找到p,q的最近公共祖先
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) return null;
//若p,q均在root右子树,则去右子树递归
if (p.val > root.val && q.val > root.val) {
return lowestCommonAncestor(root.right, p, q);
}
//若p,q均在root左子树,则去左子树递归
if (p.val < root.val && q.val < root.val) {
return lowestCommonAncestor(root.left, p, q);
}
//其他所有情况,最近公共祖先均为root。如:1)p,q分别在左右子树; 2)p指向root; 3)q指向root
return root;
}
}