LeetCode 236. 二叉树的最近公共祖先

236. 二叉树的最近公共祖先

解题思路

1. 如果当前节点是空节点，返回 null
2. 如果 p 或者 q 为根节点，返回根节点
3. 分别递归查找左右子树中的 p、q
4. 若左右子树都非空，说明 p、q 分居当前节点两侧，当前节点就是最近公共祖先，返回它
5. 若只有一侧非空，说明 p、q 都在这一侧，返回该侧的查找结果；两侧都空则返回 null

参考代码

class Solution {
    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);
        if(left != null && right != null) {
            return root;
        }
        return left == null ? right : left;
    }
}