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

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

题目描述

解题思路

这道题目的思路主要就是判断 p 和 q 的左右位置：

• 若 p，q 中有一个指向 root，则最近公共祖先就是root
• 若 p，q 分别在 root 树的两侧，则最近公共祖先就是root
• 若 p，q 均在 root 树的同一边子树中，则最近公共祖先就是 该子树的最近公共祖先

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