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

class Solution {
    /**
    思路: 后序遍历
        如果在一颗子树上找到 p 或 q , 直接返回当前的值, 如果一个节点, 
    他的左右子树分别存在 p 和 q, 那么该节点一定是 p 和 q 的最近公共祖先
    对于特殊情况: p 和 q 的最低公共祖先是他自己时, 该过程同样是成立的
     */
    public TreeNode lowestCommonAncestor(TreeNode root, 
                                        TreeNode p, TreeNode q) {
        if (root == null)
            return null;
        // 找到 p 或 q, 直接返回
        if (root == p || root == q) 
            return root;
        TreeNode l = lowestCommonAncestor(root.left, p, q);
        TreeNode r = lowestCommonAncestor(root.right, p, q);
        // 如果当前节点的左右子树分别存在 q 和 p , 那么说明当前节点就是
        // p 和 q 的最低公共祖先
        if ((l == q && r == p) || (l == p && r == q)) {
            return root;
        } else if (l != null) {
            return l;
        } else if (r != null) {
            return r;
        } else {
            return null;
        }
    }
}