剑指offer 68 - I. 二叉搜索树的最近公共祖先

剑指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;
    }
}

利用二叉搜索树的性质（推荐）

参考题解：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-lcof/solution/mian-shi-ti-68-i-er-cha-sou-suo-shu-de-zui-jin-g-7/

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;
    }
}